Update app.py

app.py

@@ -153,7 +153,7 @@ def create_tags(image, threshold):
     tensor = transform(img).unsqueeze(0)
 
     with torch.no_grad():
-        probits = model(tensor)
+        probits = model(tensor).squeeze()
         indices = torch.where(probits > threshold)[0]
         values = probits[indices]