Update app.py

app.py

@@ -44,6 +44,9 @@ prompt_builder = vlm.get_prompt_builder()
 def bot_streaming(message, history, temperature, top_k, max_new_tokens):
     if len(history) == 0:
         prompt_builder.prompt, prompt_builder.turn_count = "", 0
+
+    image = None
+    
     if message["files"]:
         image = message["files"][-1]["path"]
     else:
@@ -53,7 +56,8 @@ def bot_streaming(message, history, temperature, top_k, max_new_tokens):
             if type(hist[0])==tuple:
                 image = hist[0][0]
   
-    image = Image.open(image).convert("RGB")
+    if image is not None:
+        image = Image.open(image).convert("RGB")
     
     prompt_builder.add_turn(role="human", message=message['text'])
     prompt_text = prompt_builder.get_prompt()