Upload app.py

app.py

@@ -12,5 +12,9 @@ def caption_image(raw_image):
     out = model.generate(**inputs)
     return processor.decode(out[0], skip_special_tokens=True)
 
-demo = gr.Interface(fn=caption_image, inputs="image", outputs="caption")
+outputs = [
+           gr.outputs.Textbox(label="Caption, including detected generator (if applicable)"),
+          ] 
+
+demo = gr.Interface(fn=caption_image, inputs="image", outputs=outputs)
 demo.launch()