Update app.py

app.py

@@ -174,7 +174,7 @@ class main():
             
  
     @torch.no_grad()
-    @spaces.GPU(duration=1000)
+    @spaces.GPU(duration=120)
     def inference(self, prompt, negative_prompt, guidance_scale, ddim_steps, seed):
         device = self.device
         self.unet.to(device)
@@ -314,7 +314,7 @@ class main():
         return image
 
     @torch.no_grad()
-    @spaces.GPU(duration=1000)
+    @spaces.GPU(duration=120)
     def sample_then_run(self):
         self.unet = UNet2DConditionModel.from_pretrained(
             pretrained_model_name_or_path, subfolder="unet", revision=revision