Update app.py

app.py

@@ -33,19 +33,22 @@ def clean_response(response: str) -> str:
     return response
 
 def chat_with_model(image_path=None, text_query=None, history=None):
-    # Modify this function to use 'history' if your model requires it
-    query_elements = []
-    if image_path:
-        query_elements.append({'image': image_path})
-    if text_query:
-        query_elements.append({'text': text_query})
-    # Add history processing here if needed
+
+    default_image_path = 'path/to/default/image.jpg'  
+    default_text = 'No text provided'
+    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
+    
 def draw_boxes(image_path, response):
     image = Image.open(image_path)
     draw = ImageDraw.Draw(image)