Update app.py

app.py

@@ -9,9 +9,9 @@ processor = AutoProcessor.from_pretrained("microsoft/kosmos-2-patch14-224")
 
 # The original Kosmos-2 demo saves the image first then reload it. For some images, this will give slightly different image input and change the generation outputs.
 
-prompt = "{question}"
+#prompt = "{question}"
 
-def describe_image(image_path, question):
+def describe_image(image_path, question : str):
   inputs = processor(text=question, images=image_path, return_tensors="pt")
 
   generated_ids = model.generate(
@@ -34,7 +34,7 @@ def describe_image(image_path, question):
 
 import gradio as gr
 
-gr_app = gr.Interface(fn=describe_image, inputs=[gr.Image(label="Upload an image for description", type='pil'), "text"],
+gr_app = gr.Interface(fn=describe_image, inputs=[gr.Image(label="Upload an image for description", type='pil'), gr.Textbox(label="Ask a question about the image")],
                       outputs=[gr.Textbox(label="Image description")], title="App for image description")
 
 if __name__ == "__main__":