Update app.py

app.py

@@ -57,7 +57,9 @@ title = "Clothing Identifier"
 
 def classify_image(img):
     pred, idx, probs = learn.predict(img)
-    return dict(zip(categories, map(float, probs)))
+    highest_prob_index = probs.argmax()  # Find the index of the highest probability
+    return categories[highest_prob_index]  # Return the class with the highest probability
+
 
 
 iface = gr.Interface(