Update app.py

app.py

@@ -19,7 +19,7 @@ def process_image(image, prompt):
        output = model.generate(**inputs, max_new_tokens=10)
       
        # Decode and return the output
-       decoded_output = processor.batch_decode(generated_ids, skip_special_tokens=True)[0].strip()
+       decoded_output = processor.batch_decode(output, skip_special_tokens=True)[0].strip()
       
        # Return the answer (exclude the prompt part from output)
        return decoded_output[len(prompt):]
@@ -34,6 +34,6 @@ inputs = [
 ]
 outputs = gr.Textbox(label="Answer")
 # Create the Gradio app
-demo = gr.Interface(fn=process_image, inputs=inputs, outputs=decoded_output, title="Visual Question Answering", description="Upload an image and ask questions to get answers.")
+demo = gr.Interface(fn=process_image, inputs=inputs, outputs=outputs, title="Visual Question Answering", description="Upload an image and ask questions to get answers.")
 # Launch the app
 demo.launch()