Fix: Gradcam issue

app.py

@@ -53,7 +53,7 @@ def inference_wrapper(image, alpha, top_k, target_layer):
     """
     try:
         if image is None:
-            return {"error": "No image provided"}, None
+            return {"No image provided": 1.0}, None
             
         results = inference(
             image, 
@@ -65,13 +65,20 @@ def inference_wrapper(image, alpha, top_k, target_layer):
         )
         
         if results is None:
-            return {"error": "Processing failed"}, None
+            return {"Processing failed": 1.0}, None
             
         return results
         
     except Exception as e:
-        print(f"Error in inference: {str(e)}")
-        return {"error": str(e)}, None
+        error_msg = str(e)
+        print(f"Error in inference: {error_msg}")
+        
+        # Handle GPU quota error specifically
+        if "GPU quota" in error_msg:
+            return {"GPU quota exceeded - Please try again later": 1.0}, None
+        
+        # Handle other errors
+        return {"Error: " + error_msg: 1.0}, None
 
 
 def main():
@@ -189,9 +196,9 @@ def main():
                 )
 
             # Configure queue
-            demo.queue()
+            demo.queue(concurrency_count=1)  # Limit concurrent processing
 
-            # Launch with minimal parameters
+            # Launch with compatible parameters
             demo.launch(
                 server_name="0.0.0.0",
                 server_port=7860,