Update app.py

app.py

@@ -72,4 +72,4 @@ async def predict_endpoint(input: fastapi.UploadFile):
     # Return the top N class names and their probabilities in JSON format
     
 
-    return {"predictions": class_names[prediction]}
+    return {"predictions": prediction}