Update app.py

app.py

@@ -28,18 +28,17 @@ pipe.to("cuda")
 
 # Definição dos LoRA e Trigger Words
 lora_models = {
-    "AndroFlux": {
+    "vgn": {
         "repo": "vcollos/vgn",
         "weights": "vgn.safetensors",
         "trigger_word": ""  # Sem trigger word específica
     },
-    "VitorCollos": {
+    "Nanda": {
         "repo": "vcollos/Nanda",
         "weights": "lora.safetensors",
-        "trigger_word": "A photo of Vitor, RAW photo, (hyperrealistic portrait:1.3) of a [man/woman], (detailed eyes:1.2), (skin texture:1.4), (natural lighting:1.1), (soft shadows:1.1), (intricate hair details:1.3), (film grain:0.8), (8k:1.2), (depth of field:1.1), (sharp focus:1.1),"
+        "trigger_word": "A photo of Nanda, RAW photo, (hyperrealistic portrait:1.3) of a [man/woman], (detailed eyes:1.2), (skin texture:1.4), (natural lighting:1.1), (soft shadows:1.1), (intricate hair details:1.3), (film grain:0.8), (8k:1.2), (depth of field:1.1), (sharp focus:1.1),"
     }
 }
-
 # Carrega os LoRAs
 for name, details in lora_models.items():
     try:
@@ -73,28 +72,6 @@ def run_lora(prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora
         seed = random.randint(0, MAX_SEED)
     generator = torch.Generator(device="cuda").manual_seed(seed)
 
-    # Moderação de texto
-    moderation_client = client_gradio("duchaba/Friendly_Text_Moderation")
-    result = moderation_client.predict(
-        msg=f"{prompt}",
-        safer=0.02,
-        api_name="/fetch_toxicity_level"
-    )
-
-    if float(json.loads(result[1])['sexual_minors']) > 0.03:
-        print('🔴 Conteúdo não permitido')
-        supabase.table("requests").insert({
-            "prompt": prompt,
-            "cfg_scale": cfg_scale,
-            "steps": steps,
-            "randomized_seed": randomize_seed,
-            "seed": seed,
-            "lora_scale_1": lora_scale_1,
-            "lora_scale_2": lora_scale_2,
-            "moderated": 'true'
-        }).execute()
-        raise gr.Error("🚫 Requisição não autorizada!")
-
     # Aplica os adaptadores LoRA corretamente
     pipe.set_adapters([selected_lora], adapter_weights=[1.0])
 
@@ -139,7 +116,8 @@ 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("#Collos Image Generator")
+    gr.Markdown("# vgn Image Generator")
+    
     with gr.Row():
         with gr.Column(scale=2):
             prompt = gr.TextArea(label="Prompt", placeholder="Digite um prompt (máx 77 caracteres)", lines=3)
@@ -147,19 +125,26 @@ with gr.Blocks(theme=gr_theme) as app:
             cfg_scale = gr.Slider(label="CFG Scale", minimum=1, maximum=20, step=0.5, value=3.5)
             steps = gr.Slider(label="Steps", minimum=1, maximum=100, step=1, value=25)
             width = gr.Slider(label="Width", minimum=256, maximum=1024, step=64, value=768)  
-            height = gr.Slider(label="Height", minimum=256, maximum=1024, step=64, value=768)  
+            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)
-            selected_lora = gr.Dropdown(label="Selecionar LoRA", choices=["AndroFlux", "VitorCollos"], value="AndroFlux")
+
+            # 🔥 Certificando que os sliders estão dentro do bloco correto
+            lora_scale_1 = gr.Slider(label="LoRA Scale (Vga)", minimum=0, maximum=1, step=0.01, value=0.5)
+            lora_scale_2 = gr.Slider(label="LoRA Scale (Nanda)", minimum=0, maximum=1, step=0.01, value=1)
+
+            selected_lora = gr.Dropdown(label="Selecionar LoRA", choices=["vgn", "Nanda"], value="Nanda")
+
         with gr.Column(scale=2):
             result = gr.Image(label="Generated Image")
-            gr.Markdown("Gere imagens usando Androflux LoRA e um prompt de texto.")
+            gr.Markdown("Gere imagens usando vgn LoRA e um prompt de texto.")
 
+    # 🔥 Agora os sliders são usados corretamente
     generate_button.click(
         run_lora,
-        inputs=[prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale_1, lora_scale_2, selected_lora],  # Correção aqui
+        inputs=[prompt, cfg_scale, steps, randomize_seed, seed, width, height, lora_scale_1, lora_scale_2, selected_lora],
         outputs=[result, seed],
     )
 
 app.queue()
-app.launch(share=True)  # `share=True` cria um link público
+app.launch(share=True)