Update app.py

app.py

@@ -88,7 +88,7 @@ def bot_streaming(message, history):
     image = Image.open(image).convert('RGB')
     image_tensor = process_images([image], image_processor, model.config)[0]
 
-    inputs = tokenizer_image_token(prompt, tokenizer, IMAGE_TOKEN_INDEX, return_tensors='pt')
+    inputs = tokenizer_image_token(prompt, tokenizer, IMAGE_TOKEN_INDEX, return_tensors='pt').unsqueeze(0)
 
     print('inputs', inputs)