bugfix

app.py

@@ -53,12 +53,12 @@ class calculateDuration:
 
 
 @spaces.GPU()
-@torch.inference_mode()
-@torch.autocast(device_type="cuda", dtype=torch.bfloat16)
 def process_image(image_input, image_url, task_prompt, text_prompt=None, dilate=0, merge_masks=False, return_rectangles=False, progress=gr.Progress(track_tqdm=True)) -> Optional[Image.Image]:
+    
     if not image_input:
         gr.Info("Please upload an image.")
         return None
+    
     if not task_prompt:
         gr.Info("Please enter a task prompt.")
         return None
@@ -70,6 +70,7 @@ def process_image(image_input, image_url, task_prompt, text_prompt=None, dilate=
             response.raise_for_status()
             image_input = PIL.Image.open(BytesIO(response.content))
             print("fetch image success")
+            
     # start to parse prompt
     with calculateDuration("run_florence_inference"):
         _, result = run_florence_inference(