Spaces:

ikraamkb
/

qtAnswering

Running

ikraamkb commited on Apr 19

Commit

6b9ec82

verified ·

1 Parent(s): 5e640af

Update appImage.py

Files changed (1) hide show

appImage.py CHANGED Viewed

@@ -104,7 +104,7 @@ def answer_question_from_image(image, question):
         elif mode == "caption":
             image_tensor = caption_processor(images=image, return_tensors="pt").pixel_values
-            generated_ids = caption_model.generate(image_tensor, max_length=64)
             answer = caption_processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
         else:

         elif mode == "caption":
             image_tensor = caption_processor(images=image, return_tensors="pt").pixel_values
+            generated_ids = caption_model.generate(image_tensor, max_new_tokens=64)
             answer = caption_processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
         else: