Update app.py

app.py

@@ -14,7 +14,7 @@ question = "What does the label 15 represent? (1) lava (2) core (3) tunnel (4) a
 
 inputs = processor(images=image, text=question, return_tensors="pt")
 
-predictions = model.generate(**inputs)
+predictions = model.generate(**inputs,max_new_tokens= 50)
 # print(processor.decode(predictions[0], skip_special_tokens=True))
 
 
@@ -36,7 +36,7 @@ def visual_qna():
     img = load_image()
     if img:
         if query := st.chat_input("Enter your message"):
-            response = model(question=query, image=img,max_new_tokens= 50)
+            response = model(question=query, image=img)
             with st.chat_message("assistant"):
                 st.write(response)
     else: