iamrobotbear commited on
Commit
87b83c0
·
1 Parent(s): fcd98ee

add labels to output?

Browse files
Files changed (1) hide show
  1. app.py +11 -9
app.py CHANGED
@@ -41,15 +41,17 @@ def blip2_interface(image, prompted_caption_text, vqa_question, chat_context):
41
 
42
  return image_caption, prompted_caption, vqa_answer, chat_response
43
 
44
- # Define Gradio input and output components
45
- image_input = gr.Image(type="numpy")
46
- prompted_caption_input = gr.Textbox()
47
- vqa_question_input = gr.Textbox()
48
- chat_context = gr.Textbox()
49
- image_caption_result = gr.Textbox()
50
- prompted_caption_result = gr.Textbox()
51
- vqa_answer = gr.Textbox()
52
- chat_response = gr.Textbox()
 
 
53
 
54
  # Create Gradio interface
55
  iface = gr.Interface(
 
41
 
42
  return image_caption, prompted_caption, vqa_answer, chat_response
43
 
44
+ # Define Gradio input components
45
+ image_input = gr.inputs.Image(type="numpy", label="Image Input")
46
+ prompted_caption_input = gr.inputs.Textbox(label="Prompted Caption Text")
47
+ vqa_question_input = gr.inputs.Textbox(label="VQA Question")
48
+ chat_context = gr.inputs.Textbox(label="Chat Context")
49
+
50
+ # Define Gradio output components with labels corresponding to the inputs
51
+ image_caption_result = gr.outputs.Textbox(label="Image Caption")
52
+ prompted_caption_result = gr.outputs.Textbox(label="Prompted Image Caption")
53
+ vqa_answer = gr.outputs.Textbox(label="VQA Answer")
54
+ chat_response = gr.outputs.Textbox(label="Chat Response")
55
 
56
  # Create Gradio interface
57
  iface = gr.Interface(