Small fix for ZeroGPU

app.py

@@ -26,7 +26,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 weight_dtype = torch.bfloat16
 task = None
 
-@spaces.GPU
+
 def load_pipeline():
     global pipeline, device, weight_dtype, task
     noise_scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(
@@ -55,7 +55,7 @@ def fn(image_path):
     pipeline.set_progress_bar_config(disable=True)
     with nullcontext():
         _, output_vis, _ = process_single_image(
-            image_path, pipeline, 
+            image_path, pipeline,
             task_name=task,
             device=device,
             num_inference_steps=10,