Update app.py

app.py

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