Update app.py

app.py

@@ -2,19 +2,18 @@ import gradio as gr
 from diffusers import StableDiffusionImg2ImgPipeline
 import torch
 
-# فقط CPU
+# Load model
 pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
     "runwayml/stable-diffusion-v1-5",
-    torch_dtype=torch.float32,  # float16 روی CPU نمی‌شه
-    safety_checker=None,
-    requires_safety_checker=False
+    torch_dtype=torch.float32,
+    safety_checker=None
 )
 
-def generate(image, prompt, negative_prompt="", steps=20, strength=0.3, cfg=7.0, lora_path=None, lora_strength=0.6):
+# Load LoRA directly from file (no upload needed)
+pipe.load_lora_weights("Grool LORA.safetensors")
+
+def generate(image, prompt, negative_prompt="", steps=20, strength=0.3, cfg=7.0):
     try:
-        if lora_path:
-            pipe.load_lora_weights(lora_path)
-        
         result = pipe(
             prompt=prompt,
             negative_prompt=negative_prompt,
@@ -23,26 +22,23 @@ def generate(image, prompt, negative_prompt="", steps=20, strength=0.3, cfg=7.0,
             strength=strength,
             guidance_scale=cfg
         ).images[0]
-        
         return result
     except Exception as e:
         return f"Error: {str(e)}"
 
 with gr.Blocks() as demo:
-    gr.Markdown("## NSFW Image-to-Image (CPU Only)")
+    gr.Markdown("## NSFW Face Swap (LoRA Built-in)")
     with gr.Row():
         with gr.Column():
             input_img = gr.Image(type="pil", label="Upload Face Photo")
-            prompt = gr.Textbox(label="Prompt", lines=3, value="nude girl in bedroom, wearing tiny lace thong, small vulva visible, face locked to input image")
-            neg_prompt = gr.Textbox(label="Negative", value="deformed, large vulva, child")
-            steps = gr.Slider(10, 30, 20, label="Steps (کم = سریع‌تر)")
-            strength = gr.Slider(0.1, 0.5, 0.3, label="Strength")
-            cfg = gr.Slider(5, 10, 7.0, label="CFG")
-            lora_file = gr.File(label="Upload LoRA")
-            btn = gr.Button("Generate (30–60s)")
+            prompt = gr.Textbox(label="Prompt", lines=3, value="photorealistic, nude girl on bed, tiny lace thong, small pink vulva visible, wet, face locked to input image")
+            neg_prompt = gr.Textbox(label="Negative", value="large vulva, deformed, plastic, child")
+            steps = gr.Slider(15, 30, 20, label="Steps")
+            strength = gr.Slider(0.2, 0.5, 0.3, label="Strength")
+            btn = gr.Button("Generate (30-60s)")
         with gr.Column():
             output = gr.Image(label="Result")
     
-    btn.click(generate, [input_img, prompt, neg_prompt, steps, strength, cfg, lora_file], output)
+    btn.click(generate, [input_img, prompt, neg_prompt, steps, strength], output)
 
 demo.launch()