Add application file

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)

requirements.txt

@@ -1,3 +1,4 @@
+transformers
 diffusers
 gradio
 torch