Update app.py

app.py

@@ -171,14 +171,14 @@ class main():
 
 
     @torch.no_grad()
-    @spaces.GPU
+    @spaces.GPU(duration=1000)
     def sample_model(self):
         self.unet, _, _, _, _ = load_models(self.device)
         self.network = sample_weights(self.unet, self.proj, self.mean, self.std, self.v[:, :1000], self.device, factor = 1.00)
             
  
     @torch.no_grad()
-    @spaces.GPU
+    @spaces.GPU(duration=1000)
     def inference(self, prompt, negative_prompt, guidance_scale, ddim_steps, seed):
         device = self.device
         self.unet.to(device)