Bugfix

app.py

@@ -45,7 +45,7 @@ def predict(image: Image.Image):
 		preds = model(batch)
 		tag_preds = preds['tags'].sigmoid().cpu()
 	
-	return {top_tags[i]: tag_preds[i] for i in range(len(top_tags))}
+	return {top_tags[i]: tag_preds[0][i] for i in range(len(top_tags))}
 
 
 print("Downloading model...")