Update app.py

app.py

@@ -1,22 +1,32 @@
 import gradio as gr
 from handler import EndpointHandler
-import os
+import io
 
 # Model loading
 handler = EndpointHandler(".")
 
 def predict(image):
     try:
-        # Image input ko handler me bhejna
-        result = handler({"inputs": image})
-        return result["prediction"]
+        # 1. PIL Image ko Bytes me convert karein
+        buf = io.BytesIO()
+        image.save(buf, format='PNG')
+        image_bytes = buf.getvalue()
+
+        # 2. Bytes ko handler me bhejna (Jaise apka local code chahta hai)
+        result = handler(image_bytes)
+        
+        # Agar result dictionary hai toh prediction key lo, varna direct return
+        if isinstance(result, dict):
+            return result.get("prediction", "No Prediction")
+        return str(result)
+        
     except Exception as e:
-        return str(e)
+        return f"Error: {str(e)}"
 
-# Gradio Interface (API provide karega)
+# Interface
 demo = gr.Interface(
     fn=predict, 
-    inputs=gr.Image(type="pil"), 
+    inputs=gr.Image(type="pil"), # Gradio PIL image deta hai
     outputs="text",
     title="Master Brain API"
 )