Update app.py

app.py

@@ -1,11 +1,20 @@
+# ----------------------------------------------------------------------
+# 0️⃣ Install the exact versions we need (run once at start‑up)
+# ----------------------------------------------------------------------
+# 1️⃣ Install a nightly torch < 2.9 (as you already did)
+# 2️⃣ Upgrade huggingface‑hub to the version required by the Qwen pipeline
+# 3️⃣ Install a recent transformers build (needed for Qwen2.5‑VL)
 import os
-# PyTorch 2.8 (temporary hack)
 os.system(
-    'pip install --upgrade --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu126 "torch<2.9" spaces'
+    'pip install --upgrade --pre --extra-index-url https://download.pytorch.org/whl/nightly/cu126 '
+    '"torch<2.9" '
+    '"huggingface-hub>=1.0.0rc6" '
+    '"transformers>=4.40.0" '
+    'spaces -q'
 )
 
 # ----------------------------------------------------------------------
-# 1️⃣ Imports
+# 1️⃣ Imports (now safe because the correct packages are installed)
 # ----------------------------------------------------------------------
 import spaces
 import gradio as gr
@@ -57,7 +66,7 @@ def translate_albanian_to_english(text: str, language: str = "en"):
     raise gr.Error("Translation failed. Please try again.")
 
 # ----------------------------------------------------------------------
-# 4️⃣ Model & Scheduler Loading (run once at startup)
+# 4️⃣ Scheduler & Model Loading (run once at startup)
 # ----------------------------------------------------------------------
 scheduler_config = {
     "base_image_seq_len": 256,
@@ -85,7 +94,7 @@ pipeline = QwenImageEditPlusPipeline.from_pretrained(
 pipeline.to("cuda")
 pipeline.set_progress_bar_config(disable=None)
 
-# Load Lightning LoRA and fuse
+# Load Lightning LoRA and fuse it (speed‑up)
 pipeline.load_lora_weights(
     "lightx2v/Qwen-Image-Lightning",
     weight_name="Qwen-Image-Lightning-8steps-V2.0-bf16.safetensors",
@@ -110,7 +119,7 @@ def edit_images(image1, image2, prompt):
     prompt_en = translate_albanian_to_english(prompt.strip(), language="en")
     prompt_final = prompt_en + QUALITY_PROMPT
 
-    # Ensure PIL Images
+    # Ensure we have PIL Images
     if not isinstance(image1, Image.Image):
         image1 = Image.fromarray(image1)
     if not isinstance(image2, Image.Image):
@@ -430,7 +439,7 @@ def create_demo():
 
         with gr.Row(elem_id="general_items"):
             gr.Markdown("# Image Fusion")
-            gr.Markdown("Blend images together guided by prompt description.", elem_id="subtitle")
+            gr.Markdown("Blend images together guided by a prompt description.", elem_id="subtitle")
             with gr.Column(elem_id="input_column"):
                 image1_input = gr.Image(
                     label="First Image",