Update app.py

app.py

@@ -152,6 +152,7 @@ def generate_random_4_digit_string():
 def run_lora(prompt, image_url, lora_strings_json, image_strength,  cfg_scale, steps, randomize_seed, seed, width, height, upload_to_r2, account_id, access_key, secret_key, bucket, progress=gr.Progress(track_tqdm=True)):
     print("run_lora", prompt, lora_strings_json, cfg_scale, steps, width, height)
     gr.Info("Starting process")
+    txt2img_pipe.to("cuda")
     torch.cuda.empty_cache() 
     img2img_model = False
     orginal_image = None