Update app.py

app.py

@@ -36,9 +36,15 @@ def infer(
         torch_dtype = torch.float16
     else:
         torch_dtype = torch.float32
-        
-    pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch_dtype)
-    pipe = pipe.to(device)
+
+    if model == "Ramzes":
+        pipe = StableDiffusionPipeline.from_pretrained("stable-diffusion-v1-5/stable-diffusion-v1-5", torch_dtype=torch_dtype)
+        pipe.unet = PeftModel.from_pretrained(pipe_v5_2.unet, "Bordoglor/Ramzes_adapter_sd_v1.5/unet")
+        pipe.text_encoder = PeftModel.from_pretrained(pipe_v5_2.text_encoder, "Bordoglor/Ramzes_adapter_sd_v1.5/text_encoder")
+        pipe = pipe.to(device)
+    else:          
+        pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch_dtype)
+        pipe = pipe.to(device)
 
     image = pipe(
         prompt=prompt,
@@ -66,7 +72,7 @@ with gr.Blocks(css=css) as demo:
 
         with gr.Row():
             model = gr.Dropdown(
-                choices=["stabilityai/sdxl-turbo", "CompVis/stable-diffusion-v1-4"],
+                choices=["stabilityai/sdxl-turbo", "CompVis/stable-diffusion-v1-4", "Ramzes"],
                 value=model_repo_id,
                 label="Model",
                 info="Choose which diffusion model to use"
@@ -157,4 +163,4 @@ with gr.Blocks(css=css) as demo:
     )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()