Update app.py

app.py

@@ -1,16 +1,15 @@
 # ----------------------------------------------------------------------
 # 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)
+#  • PyTorch 2.9 (nightly) + matching torchvision/torchaudio
+#  • transformers ≥ 4.44 (contains Qwen2_5_VLForConditionalGeneration)
+#  • huggingface‑hub ≥ 1.0.0rc6  (required by the Qwen pipeline)
+#  • spaces (already used)
 import os
 os.system(
     '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'
+    '"torch==2.9.0" "torchvision==0.24.0" "torchaudio==2.9.0" '
+    '"transformers>=4.44" "huggingface-hub>=1.0.0rc6" spaces -q'
 )
 
 # ----------------------------------------------------------------------
@@ -119,7 +118,7 @@ def edit_images(image1, image2, prompt):
     prompt_en = translate_albanian_to_english(prompt.strip(), language="en")
     prompt_final = prompt_en + QUALITY_PROMPT
 
-    # Ensure we have PIL Images
+    # Ensure PIL Images
     if not isinstance(image1, Image.Image):
         image1 = Image.fromarray(image1)
     if not isinstance(image2, Image.Image):