Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-import torch, os, json, requests
+import torch, os, json, requests, sys
 from PIL import Image
 from transformers import DonutProcessor, VisionEncoderDecoderModel, VisionEncoderDecoderConfig
 from torchvision import transforms
@@ -51,6 +51,8 @@ class OCRVQAModel(torch.nn.Module):
         # try:    
         self.donut.eval()
         with torch.no_grad():
+
+            print(type(image), type(prompt), file = sys.stderr)
             image_ids = self.processor(image, return_tensors="pt").pixel_values.to(device)
 
             question = f'<s_docvqa><s_question>{prompt}</s_question><s_answer>'
@@ -120,7 +122,6 @@ with gr.Blocks() as demo:
             image_url = gr.Textbox(lines=1, label="Image URL", placeholder="Or, paste the image URL here")
             question = gr.Textbox(lines=5, label="Question")
 
-        with gr.Column():
             ask = gr.Button(label="Get the answer")
 
         with gr.Column():