Update app.py

app.py

@@ -33,14 +33,14 @@ pipe.to("cuda")
 
 # Definição dos LoRA e Trigger Words
 lora_models = {
-    "studio": {
+    "Paula": {
         "repo": "vcollos/Paula2",
         "weights": "Paula P.safetensors",
         "trigger_word": ""  # Sem trigger word específica
     },
-    "Nanda": {
-        "repo": "vcollos/Nanda",
-        "weights": "lora.safetensors",
+    "Vivi": {
+        "repo": "vcollos/Vivi",
+        "weights": "Vivi.safetensors",
         "trigger_word": ""
     }
 }
@@ -89,21 +89,21 @@ def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora
     selected_loras = []
     adapter_weights = []
 
-    if lora_option == "studio":
-        selected_loras.append("studio")
+    if lora_option == "Paula":
+        selected_loras.append("Paula")
         adapter_weights.append(lora_scale_1)
-    elif lora_option == "Nanda":
-        selected_loras.append("Nanda")
+    elif lora_option == "Vivi":
+        selected_loras.append("Vivi")
         adapter_weights.append(lora_scale_2)
     elif lora_option == "Ambos":
-        selected_loras = ["studio", "Nanda"]
+        selected_loras = ["Paula", "Vivi"]
         adapter_weights = [lora_scale_1, lora_scale_2]
 
     pipe.set_adapters(selected_loras, adapter_weights)
 
-    # Adiciona trigger words apenas se Nanda estiver ativado
-    if "Nanda" in selected_loras:
-        prompt = f"{lora_models['Nanda']['trigger_word']} {prompt}"
+    # Adiciona trigger words apenas se Vivi estiver ativado
+    if "Vivi" in selected_loras:
+        prompt = f"{lora_models['Vivi']['trigger_word']} {prompt}"
 
     # Gera a imagem com precisão de 16 bits para tentar melhorar a nitidez
     with torch.autocast("cuda"):
@@ -157,7 +157,7 @@ def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora
 # Interface Gradio
 gr_theme = os.getenv("THEME")
 with gr.Blocks(theme=gr_theme) as app:
-    gr.Markdown("# studio Image Generator")
+    gr.Markdown("# Paula Image Generator")
     
     with gr.Row():
         with gr.Column(scale=2):
@@ -169,9 +169,9 @@ with gr.Blocks(theme=gr_theme) as app:
             height = gr.Slider(label="Height", minimum=256, maximum=1024, step=64, value=1024)  
             randomize_seed = gr.Checkbox(False, label="Randomize seed")
             seed = gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=556215326)
-            lora_option = gr.Radio(["Nenhum", "studio", "Nanda", "Ambos"], label="Escolha o LoRA", value="Ambos")
-            lora_scale_1 = gr.Slider(label="LoRA Scale (studio)", minimum=0, maximum=1, step=0.01, value=0.1)
-            lora_scale_2 = gr.Slider(label="LoRA Scale (Nanda)", minimum=0, maximum=1, step=0.01, value=1)
+            lora_option = gr.Radio(["Nenhum", "Paula", "Vivi", "Ambos"], label="Escolha o LoRA", value="Ambos")
+            lora_scale_1 = gr.Slider(label="LoRA Scale (Paula)", minimum=0, maximum=1, step=0.01, value=1)
+            lora_scale_2 = gr.Slider(label="LoRA Scale (Vivi)", minimum=0, maximum=1, step=0.01, value=1)
 
         with gr.Column(scale=2):
             result = gr.Image(label="Generated Image")