Update app.py

app.py

@@ -9,7 +9,8 @@ import torch
 import os
 from PIL import Image, ImageDraw, UnidentifiedImageError
 
-model_name = "qwen/Qwen-VL-Chat"
+base_url = "https://huggingface.co/spaces/Tonic1/Official-Qwen-VL-Chat"
+model_name = "Qwen/Qwen-VL-Chat"
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True).eval()
 model.generation_config = GenerationConfig.from_pretrained(model_name, trust_remote_code=True)
@@ -36,18 +37,20 @@ def clean_response(response: str) -> str:
 
 def chat_with_model(image_path=None, text_query=None, history=None):
     default_image_path = 'https://huggingface.co/spaces/Tonic1/Official-Qwen-VL-Chat/blob/main/2saj7l.png'
-
     if image_path:
-        if os.path.isfile(image_path):
+        full_image_path = os.path.join(base_url, image_path) if base_url else image_path
+        if os.path.isfile(full_image_path):
             try:
-                with Image.open(image_path) as img:
-                    print(f"Image {image_path} opened successfully.")
-            except UnidentifiedImageError as e:
-                print(f"Error opening image: {e}")
-                image_path = default_image_path
+                with Image.open(full_image_path) as img:
+                    print(f"Image {full_image_path} opened successfully.")
+            except UnidentifiedImageError:
+                print(f"Error: The file at {full_image_path} is not a recognizable image.")
+                full_image_path = default_image_path
         else:
-            print(f"Image file does not exist at {image_path}, using default image.")
-            image_path = default_image_path
+            print(f"File not found: {full_image_path}")
+            full_image_path = default_image_path
+    else:
+        full_image_path = default_image_path
     else:
         print("No image path provided, using default image.")
         image_path = default_image_path