smellslikeml commited on
Commit
432802d
1 Parent(s): 6229c52

update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -7
app.py CHANGED
@@ -69,15 +69,16 @@ examples = [
69
  ["examples/warehouse_2.jpg", "Is the forklift taller than the shelves of boxes?"],
70
  ]
71
 
72
- image_input = gr.Image(type="pil", label="Input Image")
73
- text_input = gr.Textbox(label="Prompt")
74
  iface = gr.Interface(
75
  fn=predict,
76
- inputs=[image_input, text_input],
77
- outputs="text",
 
 
 
78
  )
79
 
80
- iface.add_component(gr.Markdown(title_and_links_markdown), "header")
81
- iface.set_examples(examples)
82
-
83
  iface.launch()
 
69
  ["examples/warehouse_2.jpg", "Is the forklift taller than the shelves of boxes?"],
70
  ]
71
 
72
+
73
+ # Create the Gradio interface with the markdown
74
  iface = gr.Interface(
75
  fn=predict,
76
+ inputs=[gr.Image(type="pil", label="Input Image"), gr.Textbox(label="Prompt")],
77
+ outputs=gr.Textbox(),
78
+ examples=examples,
79
+ title="🛸SpaceLLaVA🌋: A spatial reasoning multi-modal model",
80
+ description=title_and_links_markdown # Use description for markdown
81
  )
82
 
83
+ # Launch the Gradio app
 
 
84
  iface.launch()