Update main.py

main.py

@@ -52,6 +52,6 @@ async def predict(file: UploadFile):
     confidence = predictions[0][0]
 
     return {
-        'class': predicted_class,
+        'class': predicted_class, 
         "confidence": float(confidence)
     }