Update app.py

app.py

@@ -114,6 +114,8 @@ def run_lora(prompt, image_url, lora_strings_json, image_strength,  cfg_scale, s
     gr.Info("Starting process")
     img2img_model = False
     orginal_image = None
+    device = txt2img_pipe.device
+    
     print(device)
   
     # Set random seed for reproducibility
@@ -168,10 +170,10 @@ def run_lora(prompt, image_url, lora_strings_json, image_strength,  cfg_scale, s
         gr.Info("Start to generate images ...")
         print(device)
         # Generate image
-        txt2img_pipe.to(device)
-        generator = torch.Generator(device=device).manual_seed(seed)
+        pipe = txt2img_pipe.to(device)
+        generator = torch.Generator("cuda").manual_seed(seed)
         joint_attention_kwargs = {"scale": 1}
-        final_image = txt2img_pipe(
+        final_image = pipe(
             prompt=prompt,
             num_inference_steps=steps,
             guidance_scale=cfg_scale,