Update app.py

app.py

@@ -8,7 +8,6 @@ import os
 import logging
 import requests
 from io import BytesIO
-import json
 
 # Setup logging
 logging.basicConfig(level=logging.INFO)
@@ -79,42 +78,37 @@ def predict_from_image_url(image_url):
 
         # Interpret the result
         if predicted_class == 0:
-            json_output = json.dumps({
-                "result": "The photo shows symptoms of Fall army worm with Problem ID 126.",
+            return {
+                "result": "The photo is of Fall Army Worm with problem ID 126.",
                 "probabilities": {
                     "Fall Army Worm": f"{probabilities[0]*100:.2f}%",
                     "Phosphorus Deficiency": f"{probabilities[1]*100:.2f}%",
                     "Bacterial Leaf Blight": f"{probabilities[2]*100:.2f}%"
                 }
-            })
-            return  json_output
+            }
         elif predicted_class == 1:
-            json_output = json.dumps({
-                "result": "The photo shows symptoms of Phosphorus deficiency with Problem ID 142.",
+            return {
+                "result": "The photo shows symptoms of Phosphorus Deficiency with Problem ID 142.",
                 "probabilities": {
                     "Fall Army Worm": f"{probabilities[0]*100:.2f}%",
                     "Phosphorus Deficiency": f"{probabilities[1]*100:.2f}%",
                     "Bacterial Leaf Blight": f"{probabilities[2]*100:.2f}%"
                 }
-            })
-            return json_output
+            }
         elif predicted_class == 2:
-            json_output = json.dumps({
+            return {
                 "result": "The photo shows symptoms of Bacterial Leaf Blight with Problem ID 203.",
                 "probabilities": {
                     "Fall Army Worm": f"{probabilities[0]*100:.2f}%",
                     "Phosphorus Deficiency": f"{probabilities[1]*100:.2f}%",
                     "Bacterial Leaf Blight": f"{probabilities[2]*100:.2f}%"
                 }
-            })
-            return json_output
+            }
         else:
-            json_output = json.dumps({"error": "Unexpected class prediction."})
-            return json_output
+            return {"error": "Unexpected class prediction."}
 
     except Exception as e:
-            json_output = json.dumps({"error": str(e)})
-            return json_output
+        return {"error": str(e)}
 
 # Gradio interface
 demo = gr.Interface(