Update app.py

app.py

@@ -16,8 +16,10 @@ def read_content(file_path: str) -> str:
 
 
 def predict(prompt, negative_prompt, guidance_scale, num_inference_steps,model, scheduler, lora, lora_weight):
-    pipeline = diffusers.DiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V6.0_B1_noVAE", vae=vae).to("cuda")
-    pipeline.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
+    pipeline = diffusers.DiffusionPipeline.from_pretrained("SG161222/RealVisXL_V4.0", vae=vae).to("cuda")
+    if model = 'Realistic_V6.0':
+        pipeline = diffusers.DiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V6.0_B1_noVAE", vae=vae).to("cuda")
+        pipeline.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
     if model == "Realistic_V5.1":
         pipeline = diffusers.DiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V5.1_noVAE", vae=vae).to("cuda")
     if model == "Realistic_V5.0":
@@ -124,8 +126,8 @@ with image_blocks as demo:
                     negative_prompt = gr.Textbox(label="negative_prompt", placeholder="Your negative prompt",
                                                  info="what you don't want to see in the image")
                 with gr.Row(equal_height=True):
-                    models = ['Realistic_V6.0','Realistic_V5.1','Realistic_V5.0','EpicRealism']
-                    model = gr.Dropdown(label="Models",choices=models,value="Realistic_V6.0")
+                    models = ['RealVisXL_V4.0','Realistic_V6.0','Realistic_V5.1','Realistic_V5.0','EpicRealism']
+                    model = gr.Dropdown(label="Models",choices=models,value="RealVisXL_V4.0")
                 with gr.Row(equal_height=True):
                     schedulers = ["DEISMultistepScheduler", "HeunDiscreteScheduler", "EulerDiscreteScheduler",
                                   "DPMSolverMultistepScheduler", "DPMSolverMultistepScheduler-Karras",
@@ -135,7 +137,7 @@ with image_blocks as demo:
                 with gr.Row(equal_height=True):
                     loras = ['None','add_detail','nayanthara','shobita','surya','vijay','saipallavi']
                     lora = gr.Dropdown(label='Lora', choices=loras, value="None")
-                    lora_weight = gr.Number(value=0.5, minimum=0, maximum=1, step=0.01, label="Lora Weights")
+                    lora_weight = gr.Number(value=0, minimum=0, maximum=1, step=0.01, label="Lora Weights")
             with gr.Row(equal_height=True):
                 btn = gr.Button("Generate", elem_id="run_button")