Update app.py

app.py

@@ -162,10 +162,7 @@ GPU_DURATION_OPTIONS = {
     "Long (140s)": 140,
 }
 
-def set_gpu_duration(duration_choice):
-    os.environ["GPU_DURATION"] = str(GPU_DURATION_OPTIONS[duration_choice])
-
-@spaces.GPU(duration=int(os.getenv("GPU_DURATION", "60")))  # Dynamic duration
+@spaces.GPU(duration=int(os.getenv("GPU_DURATION", "0")))  # Dynamic duration
 def generate(
     model_choice: str,
     prompt: str,
@@ -185,7 +182,7 @@ def generate(
     global models
     pipe = models[model_choice]
     seed = int(randomize_seed_fn(seed, randomize_seed))
-    generator = torch.Generator(device='cpu').manual_seed(seed)
+    generator = torch.Generator(device='cuda').manual_seed(seed)
 
     prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)
 
@@ -269,6 +266,9 @@ def generate_cpu(
     image_paths = [save_image(img) for img in images]
     return image_paths, seed
 
+def set_gpu_duration(duration_choice):
+    os.environ["GPU_DURATION"] = str(GPU_DURATION_OPTIONS[duration_choice])
+
 def load_predefined_images1():
     predefined_images1 = [
         "assets/7.png",