SFP commited on
Commit
171063f
1 Parent(s): b92b23b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -28,6 +28,7 @@ def caption(file, ext):
28
  return chat_completion.choices[0].message.content
29
  def image_predict(image):
30
  return caption(image, "png")
31
-
32
- iface = gr.Interface(image_predict, inputs=gr.Image(type="pil"), outputs="label", flagging_options=[])
 
33
  iface.launch()
 
28
  return chat_completion.choices[0].message.content
29
  def image_predict(image):
30
  return caption(image, "png")
31
+ title = "Generating Captions For Images Using ImCap v1"
32
+ description = "ImCap v1 uses a varity of image caption genorators, then plugs it into LLaMA 2 to get a final caption"
33
+ iface = gr.Interface(image_predict, inputs=gr.Image(type="pil"), outputs="label", flagging_options=[] title=title, description=description,)
34
  iface.launch()