Update app.py

app.py

@@ -10,6 +10,10 @@ import numpy as np
 import time
 import zipfile
 import os
+import requests
+from urllib.parse import urlparse
+import tempfile
+import shutil
 
 # Description for the app
 DESCRIPTION = """## Qwen Image Hpc/."""
@@ -44,6 +48,45 @@ aspect_ratios = {
     "3:4": (1140, 1472)
 }
 
+def load_lora_opt(pipe, lora_input):
+    lora_input = lora_input.strip()
+    if not lora_input:
+        return
+
+    # If it's just an ID like "author/model"
+    if "/" in lora_input and not lora_input.startswith("http"):
+        pipe.load_lora_weights(lora_input)
+        return
+
+    if lora_input.startswith("http"):
+        url = lora_input
+
+        # Repo page (no blob/resolve)
+        if "huggingface.co" in url and "/blob/" not in url and "/resolve/" not in url:
+            repo_id = urlparse(url).path.strip("/")
+            pipe.load_lora_weights(repo_id)
+            return
+
+        # Blob link → convert to resolve link
+        if "/blob/" in url:
+            url = url.replace("/blob/", "/resolve/")
+
+        # Download direct file
+        tmp_dir = tempfile.mkdtemp()
+        local_path = os.path.join(tmp_dir, os.path.basename(urlparse(url).path))
+
+        try:
+            print(f"Downloading LoRA from {url}...")
+            resp = requests.get(url, stream=True)
+            resp.raise_for_status()
+            with open(local_path, "wb") as f:
+                for chunk in resp.iter_content(chunk_size=8192):
+                    f.write(chunk)
+            print(f"Saved LoRA to {local_path}")
+            pipe.load_lora_weights(local_path)
+        finally:
+            shutil.rmtree(tmp_dir, ignore_errors=True)
+
 # Generation function for Qwen/Qwen-Image
 @spaces.GPU(duration=120)
 def generate_qwen(
@@ -57,6 +100,8 @@ def generate_qwen(
     num_inference_steps: int = 50,
     num_images: int = 1,
     zip_images: bool = False,
+    lora_input: str = "",
+    lora_scale: float = 1.0,
     progress=gr.Progress(track_tqdm=True),
 ):
     if randomize_seed:
@@ -64,6 +109,16 @@ def generate_qwen(
     generator = torch.Generator(device).manual_seed(seed)
     
     start_time = time.time()
+
+    pipe_qwen.unload_lora_weights()
+    use_lora = False
+    if lora_input and lora_input.strip() != "":
+        load_lora_opt(pipe_qwen, lora_input)
+        use_lora = True
+    
+    kwargs = {}
+    if use_lora:
+        kwargs["cross_attention_kwargs"] = {"scale": lora_scale}
     
     images = pipe_qwen(
         prompt=prompt,
@@ -75,6 +130,7 @@ def generate_qwen(
         num_images_per_prompt=num_images,
         generator=generator,
         output_type="pil",
+        **kwargs,
     ).images
     
     end_time = time.time()
@@ -105,6 +161,8 @@ def generate(
     num_inference_steps: int,
     num_images: int,
     zip_images: bool,
+    lora_input: str,
+    lora_scale: float,
     progress=gr.Progress(track_tqdm=True),
 ):
     final_negative_prompt = negative_prompt if use_negative_prompt else ""
@@ -119,6 +177,8 @@ def generate(
         num_inference_steps=num_inference_steps,
         num_images=num_images,
         zip_images=zip_images,
+        lora_input=lora_input,
+        lora_scale=lora_scale,
         progress=progress,
     )
 
@@ -165,6 +225,8 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
             choices=list(aspect_ratios.keys()),
             value="1:1",
         )
+    with gr.Row():
+        lora = gr.Textbox(label="qwen3 image lora (optional)", info="insert lora path")
     with gr.Accordion("Additional Options", open=False):
         use_negative_prompt = gr.Checkbox(
             label="Use negative prompt",
@@ -223,6 +285,14 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
             value=1,
         )
         zip_images = gr.Checkbox(label="Zip generated images", value=False)
+        with gr.Row(): 
+            lora_scale = gr.Slider(
+                label="LoRA Scale",
+                minimum=0,
+                maximum=2,
+                step=0.01,
+                value=1,
+            )
         
         gr.Markdown("### Output Information")
         seed_display = gr.Textbox(label="Seed used", interactive=False)
@@ -263,6 +333,8 @@ with gr.Blocks(css=css, theme="bethecloud/storj_theme") as demo:
             num_inference_steps,
             num_images,
             zip_images,
+            lora,
+            lora_scale,
         ],
         outputs=[result, seed_display, generation_time, zip_file],
         api_name="run",