Update app.py

app.py

@@ -162,7 +162,13 @@ GPU_DURATION_OPTIONS = {
     "Long (140s)": 140,
 }
 
-@spaces.GPU(duration=int(os.getenv("GPU_DURATION", "0")))  # Dynamic duration
+# Set the default GPU duration here
+DEFAULT_DURATION_CHOICE = "Medium (60s)" 
+
+# Set the GPU duration environment variable
+os.environ["GPU_DURATION"] = str(GPU_DURATION_OPTIONS[DEFAULT_DURATION_CHOICE])
+
+@spaces.GPU(duration=int(os.getenv("GPU_DURATION")))  # Dynamic duration
 def generate(
     model_choice: str,
     prompt: str,
@@ -237,7 +243,7 @@ def generate_cpu(
     pipe.to("cpu")
 
     seed = int(randomize_seed_fn(seed, randomize_seed))
-    generator = torch.Generator(device='cuda').manual_seed(seed)
+    generator = torch.Generator(device='cpu').manual_seed(seed) # cpu function seed
 
     prompt, negative_prompt = apply_style(style_selection, prompt, negative_prompt)