sd hf model

app.py

@@ -7,8 +7,8 @@ def produce_art(prompt):
     from diffusers import StableDiffusionPipeline
     import torch
 
-    model_id = "model_rita1/1600"
-    pipe = StableDiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16).to("cuda")
+    model_id = "atsyplikhin/rita_sd_model"
+    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

@@ -0,0 +1,3 @@
+diffusers
+gradio
+torch