Update app.py

app.py

@@ -3,7 +3,7 @@ import numpy as np
 import random
 
 # import spaces #[uncomment to use ZeroGPU]
-from diffusers import DiffusionPipeline
+from diffusers import DiffusionPipeline, StableDiffusionPipeline
 from peft import PeftModel, LoraConfig
 import torch
 
@@ -37,15 +37,9 @@ def infer(
         torch_dtype = torch.float16
     else:
         torch_dtype = torch.float32
-
-    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)
+        
+    pipe = DiffusionPipeline.from_pretrained(model, torch_dtype=torch_dtype)
+    pipe = pipe.to(device)
 
     image = pipe(
         prompt=prompt,
@@ -73,7 +67,7 @@ with gr.Blocks(css=css) as demo:
 
         with gr.Row():
             model = gr.Dropdown(
-                choices=["stabilityai/sdxl-turbo", "CompVis/stable-diffusion-v1-4", "Ramzes"],
+                choices=["stabilityai/sdxl-turbo", "CompVis/stable-diffusion-v1-4"],
                 value=model_repo_id,
                 label="Model",
                 info="Choose which diffusion model to use"