Update app.py

app.py

@@ -99,7 +99,7 @@ class MolmoCaptionProcessor:
             device_map='auto'
         )
 
-    @spaces.GPU(duration=120)
+    @spaces.GPU
     def generate_response(self, image: Image.Image, msg: str) -> str:
         inputs = self.processor.process(
             images=[image],
@@ -230,7 +230,7 @@ def update_prompt(images: List[Image.Image], task_type: str, custom_msg: str = N
         return "", f"Error generating captions: {str(e)}"
 
 
-@spaces.GPU(duration=120)
+@spaces.GPU
 def generate_image(images: List[Image.Image], prompt: str, negative_prompt: str, num_inference_steps: int, guidance_scale: float, 
                    denoise_mask: List[str], task_type: str, azimuth: str, elevation: str, distance: str, focal_length: float,
                    height: int = 1024, width: int = 1024, scale_factor: float = 1.0, scale_watershed: float = 1.0,