Commit
·
43bc0c8
1
Parent(s):
b24d3c5
Update app.py
Browse filesRemoved endpoints which caused errors
app.py
CHANGED
|
@@ -22,12 +22,8 @@ endpoints = [
|
|
| 22 |
"Salesforce/blip-image-captioning-base",
|
| 23 |
"nlpconnect/vit-gpt2-image-captioning",
|
| 24 |
"microsoft/git-base",
|
| 25 |
-
"microsoft/kosmos-2-patch14-224",
|
| 26 |
"microsoft/git-large-textcaps",
|
| 27 |
"microsoft/git-large-r-coco",
|
| 28 |
-
"facebook/nougat-base",
|
| 29 |
-
"Norm/nougat-latex-base",
|
| 30 |
-
"bipin/image-caption-generator",
|
| 31 |
]
|
| 32 |
|
| 33 |
|
|
@@ -77,10 +73,10 @@ def captioner(image):
|
|
| 77 |
for endpoint, result in results.items():
|
| 78 |
if "error" not in result:
|
| 79 |
caption = (
|
| 80 |
-
f"
|
| 81 |
)
|
| 82 |
else:
|
| 83 |
-
caption = f"
|
| 84 |
captions.append(caption)
|
| 85 |
return "".join(captions) # Join all captions into a single string
|
| 86 |
|
|
@@ -90,8 +86,8 @@ demo = gr.Interface(
|
|
| 90 |
fn=captioner,
|
| 91 |
inputs=[gr.Image(label="Upload image", type="pil")],
|
| 92 |
outputs=gr.Markdown(label="Captions"),
|
| 93 |
-
title="
|
| 94 |
-
description="Upload an image and see how different models
|
| 95 |
allow_flagging="never",
|
| 96 |
examples=[
|
| 97 |
"example_1.jpg",
|
|
|
|
| 22 |
"Salesforce/blip-image-captioning-base",
|
| 23 |
"nlpconnect/vit-gpt2-image-captioning",
|
| 24 |
"microsoft/git-base",
|
|
|
|
| 25 |
"microsoft/git-large-textcaps",
|
| 26 |
"microsoft/git-large-r-coco",
|
|
|
|
|
|
|
|
|
|
| 27 |
]
|
| 28 |
|
| 29 |
|
|
|
|
| 73 |
for endpoint, result in results.items():
|
| 74 |
if "error" not in result:
|
| 75 |
caption = (
|
| 76 |
+
f"## {endpoint}: \n {result[0]['generated_text']} \n\n\n"
|
| 77 |
)
|
| 78 |
else:
|
| 79 |
+
caption = f"## {endpoint}: \n Error - {result['error']} \n\n\n"
|
| 80 |
captions.append(caption)
|
| 81 |
return "".join(captions) # Join all captions into a single string
|
| 82 |
|
|
|
|
| 86 |
fn=captioner,
|
| 87 |
inputs=[gr.Image(label="Upload image", type="pil")],
|
| 88 |
outputs=gr.Markdown(label="Captions"),
|
| 89 |
+
title="Captioning Comparison",
|
| 90 |
+
description="Upload an image and see how different models describe it!",
|
| 91 |
allow_flagging="never",
|
| 92 |
examples=[
|
| 93 |
"example_1.jpg",
|