Update app.py

app.py

@@ -9,25 +9,31 @@ client = InferenceHTTPClient(
     api_key="dxkgGGHSZ3DI8XzVn29U"
 )
 
-# Prediction function
 def predict(image: Image.Image):
-    # Convert PIL image to bytes
-    image.save("temp.jpg")  # Save locally for upload
-    with open("temp.jpg", "rb") as f:
-        result = client.run_workflow(
-            workspace_name="naveen-kumar-hnmil",
-            workflow_id="detect-count-and-visualize-5",
-            images={"image": f},
-            use_cache=True
-        )
-    
-    # Get the annotated image URL from the result
-    annotated_url = result.get("visualizations", {}).get("image")
-    if annotated_url:
+    try:
+        # Save image temporarily
+        image.save("temp.jpg")
+        with open("temp.jpg", "rb") as f:
+            result = client.run_workflow(
+                workspace_name="naveen-kumar-hnmil",
+                workflow_id="detect-count-and-visualize-5",
+                images={"image": f},
+                use_cache=True
+            )
+        
+        # Extract annotated image
+        annotated_url = result.get("visualizations", {}).get("image")
+        if not annotated_url:
+            return "Error: No visualization returned from Roboflow."
+        
         response = requests.get(annotated_url, stream=True)
+        if response.status_code != 200:
+            return f"Error fetching image from: {annotated_url}"
+        
         return Image.open(response.raw)
-    else:
-        return "No annotated image returned."
+
+    except Exception as e:
+        return f"Exception: {str(e)}"
 
 # Gradio UI
 demo = gr.Interface(