Update app.py

app.py

@@ -36,20 +36,24 @@ def predict(prompt, negative_prompt, guidance_scale, num_inference_steps,model,
     
     pipeline.scheduler = scheduler.from_pretrained("emilianJR/epiCRealism", subfolder="scheduler", **add_kwargs)
     
-    if lora == "add_detail":
-        lora = "profaker/add_detail_lora"
-    if lora == "nursing_job":
-        lora = "profaker/Nursing_job_lora"
-    if lora == "nsfw_POV":
-        lora = "profaker/NSFW_POV_lora"
+    
     if lora == "nayanthara":
         lora = "profaker/Naya_lora"
+    if lora == "saipallavi":
+        lora = "profaker/saipallavi_lora"
+    if lora == "shobita":
+        lora = "profaker/Shobita_lora"
+    if lora == "surya":
+        lora = "profaker/Surya_lora"
+    if lora == "vijay":
+        lora = "profaker/Vijay_lora"
     if lora == "None":
         images = pipeline(
         prompt=prompt,
         negative_prompt=negative_prompt,
         num_inference_steps=int(num_inference_steps),
-        guidance_scale=guidance_scale
+        guidance_scale=guidance_scale,
+        clip_skip=1
         ).images[0]
         print("Prompt", prompt)
         print("Negative", negative_prompt)
@@ -129,10 +133,9 @@ with image_blocks as demo:
                     scheduler = gr.Dropdown(label="Schedulers", choices=schedulers,
                                             value="DPMSolverMultistepScheduler-Karras")
                 with gr.Row(equal_height=True):
-                    loras = ['None','add_detail', 'nursing_job', 'nsfw_POV','nayanthara']
+                    loras = ['None','add_detail','nayanthara','shobita','surya','vijay','saipallavi']
                     lora = gr.Dropdown(label='Lora', choices=loras, value="None")
-                    lora_weights = [-1, -0.5, 0, 0.5, 1]
-                    lora_weight = gr.Dropdown(label="Lora Weights", choices=lora_weights, value=0.5)
+                    lora_weight = gr.Number(value=0.5, minimum=0, maximum=1, step=0.01, label="Lora Weights")
             with gr.Row(equal_height=True):
                 btn = gr.Button("Generate", elem_id="run_button")