SFP commited on
Commit
77d1bb4
1 Parent(s): 0043576

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
  title = "Generating Captions For Images Using ImCap v1"
32
  description = """ImCap v1 uses a varity of image caption generators, then plugs it into LLaMA 2 to get a final caption.<br/>
33
  <a href="https://huggingface.co/spaces/SFP/ImCap_v1?duplicate=true"
@@ -35,5 +36,5 @@ description = """ImCap v1 uses a varity of image caption generators, then plugs
35
  <img style="margin-bottom: 0em;display: inline;margin-top: -.25em;"
36
  src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a>
37
  if there is a queue (free CPU works fine as it uses apis)."""
38
- iface = gr.Interface(image_predict, inputs=gr.Image(type="pil"), outputs="label", flagging_options=[], title=title, description=description)
39
  iface.launch(show_api=False)
 
28
  return chat_completion.choices[0].message.content
29
  def image_predict(image):
30
  return caption(image, "png")
31
+ examples = [["penguin.png"], ["stop.png"], ["stick.png"]]
32
  title = "Generating Captions For Images Using ImCap v1"
33
  description = """ImCap v1 uses a varity of image caption generators, then plugs it into LLaMA 2 to get a final caption.<br/>
34
  <a href="https://huggingface.co/spaces/SFP/ImCap_v1?duplicate=true"
 
36
  <img style="margin-bottom: 0em;display: inline;margin-top: -.25em;"
37
  src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a>
38
  if there is a queue (free CPU works fine as it uses apis)."""
39
+ iface = gr.Interface(image_predict, inputs=gr.Image(type="pil"), outputs="label", flagging_options=[], title=title, examples=examples, description=description)
40
  iface.launch(show_api=False)