Update app.py

app.py

@@ -384,12 +384,12 @@ def bot(history, temperature, top_p, max_output_tokens):
     )
     print(our_chatbot.model.device)
     print(input_ids.device)
-    print(image_tensor.device)
+    # print(image_tensor.device)
     
     generate_kwargs = dict(
         inputs=input_ids,
         streamer=streamer,
-        images=image_tensor if num_new_images > 0 else None,
+        images=image_tensor if image_tensor > 0 else None,
         do_sample=True,
         temperature=temperature,
         top_p=top_p,