Update app.py

app.py

@@ -341,22 +341,25 @@ def bot(history, temperature, top_p, max_output_tokens):
             image_list.append(load_image(f))
         else:
             raise ValueError("Invalid image file")
-    
-    image_tensor = [
-        our_chatbot.image_processor.preprocess(f, return_tensors="pt")["pixel_values"][
-            0
-        ]
-        .half()
-        .to(our_chatbot.model.device)
-        for f in image_list
-    ]
 
-
-    image_tensor = torch.stack(image_tensor)
-    image_token = DEFAULT_IMAGE_TOKEN * num_new_images
+    if num_new_images > 0:
+        image_tensor = [
+            our_chatbot.image_processor.preprocess(f, return_tensors="pt")["pixel_values"][
+                0
+            ]
+            .half()
+            .to(our_chatbot.model.device)
+            for f in image_list
+        ]
     
-    inp = text
-    inp = image_token + "\n" + inp
+    
+        image_tensor = torch.stack(image_tensor)
+        image_token = DEFAULT_IMAGE_TOKEN * num_new_images
+    
+        inp = text
+        inp = image_token + "\n" + inp
+    else:
+        inp = text
     our_chatbot.conversation.append_message(our_chatbot.conversation.roles[0], inp)
     # image = None
     our_chatbot.conversation.append_message(our_chatbot.conversation.roles[1], None)
@@ -386,7 +389,7 @@ def bot(history, temperature, top_p, max_output_tokens):
     generate_kwargs = dict(
         inputs=input_ids,
         streamer=streamer,
-        images=image_tensor,
+        images=image_tensor if num_new_images > 0 else None,
         do_sample=True,
         temperature=temperature,
         top_p=top_p,