Update app.py

app.py

@@ -6,7 +6,7 @@ import copy
 from pathlib import Path
 import secrets
 import torch
-from PIL import Image, ImageDraw
+from PIL import Image, ImageDraw, UnidentifiedImageError
 
 model_name = "qwen/Qwen-VL-Chat"
 tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
@@ -34,22 +34,24 @@ def clean_response(response: str) -> str:
     return response
 
 def chat_with_model(image_path=None, text_query=None, history=None):
-
-    default_image_path = 'path/to/default/image.jpg'  
-    default_text = 'No text provided'
+    default_image_path = None  
+    default_text = ""
     image_input = image_path if image_path else default_image_path
     text_input = text_query if text_query else default_text
     query_elements = [
         {'image': image_input},
         {'text': text_input}
     ]
-    query = tokenizer.from_list_format(query_elements)
-    tokenized_inputs = tokenizer(query, return_tensors='pt').to(device)
-    output = model.generate(**tokenized_inputs)
-    response = tokenizer.decode(output[0], skip_special_tokens=True)
-    cleaned_response = clean_response(response)
-    return cleaned_response
-    
+    try:
+        query = tokenizer.from_list_format(query_elements)
+        tokenized_inputs = tokenizer(query, return_tensors='pt').to(device)
+        output = model.generate(**tokenized_inputs)
+        response = tokenizer.decode(output[0], skip_special_tokens=True)
+        cleaned_response = clean_response(response)
+        return cleaned_response
+    except UnidentifiedImageError:
+        return "Error: Unable to process the image. Please try a different image."
+        
 def draw_boxes(image_path, response):
     image = Image.open(image_path)
     draw = ImageDraw.Draw(image)