Update app.py
Browse files
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 |
-
|
|
|
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()
|