Update app.py

app.py

@@ -8,7 +8,7 @@ def produce_art(prompt):
     import torch
 
     model_id = "atsyplikhin/rita_sd_model"
-    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16).to("cuda")
+    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)#.to("cuda")
 
     bs = 1
     images = pipe([prompt]*bs, num_inference_steps=50, guidance_scale=7.5)