Update app.py

app.py

@@ -42,7 +42,7 @@ def classify_image(inp):
                 result[label] = float(predicted_class_indices[i])
             except KeyError:
                 print(f"KeyError: Label not found for index {predicted_class_indices[i]}")
-    return predicted_class_indices    
+    return prediction