Update app.py

app.py

@@ -118,7 +118,7 @@ def infer(
     progress=gr.Progress(track_tqdm=True),
 ):
     seed = random.randint(0, MAX_SEED)
-    generator = torch.Generator(device='cpu').manual_seed(seed)
+    generator = torch.Generator(device='cuda').manual_seed(seed)
     system_prompt_rewrite = (
     "You are an AI assistant that rewrites image prompts to be more descriptive and detailed."
     )