Spaces:
Running
Running
bilgeyucel
commited on
Commit
•
e3b148f
1
Parent(s):
1e53050
Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ description = """
|
|
17 |
|
18 |
It uses [Salesforce/blip-image-captioning-base](https://huggingface.co/Salesforce/blip-image-captioning-base) model for image-to-text caption generation task.
|
19 |
|
20 |
-
For Instagrammable captions, `mistralai/Mistral-7B-Instruct-v0.
|
21 |
|
22 |
Built by [Bilge Yucel](https://twitter.com/bilgeycl) using [Haystack 2.0](https://github.com/deepset-ai/haystack) 💙
|
23 |
"""
|
@@ -54,8 +54,8 @@ with gr.Blocks(theme="soft") as demo:
|
|
54 |
image = gr.Image(type="filepath")
|
55 |
with gr.Column():
|
56 |
model_name = gr.Dropdown(
|
57 |
-
["mistralai/Mistral-7B-Instruct-v0.
|
58 |
-
value="mistralai/Mistral-7B-Instruct-v0.
|
59 |
label="Choose your model!"
|
60 |
)
|
61 |
gr.Examples(["./whale.png", "./rainbow.jpeg", "./selfie.png"], inputs=image, label="Click on any example")
|
|
|
17 |
|
18 |
It uses [Salesforce/blip-image-captioning-base](https://huggingface.co/Salesforce/blip-image-captioning-base) model for image-to-text caption generation task.
|
19 |
|
20 |
+
For Instagrammable captions, `mistralai/Mistral-7B-Instruct-v0.1` performs best, but try different models to see how they react to the same prompt.
|
21 |
|
22 |
Built by [Bilge Yucel](https://twitter.com/bilgeycl) using [Haystack 2.0](https://github.com/deepset-ai/haystack) 💙
|
23 |
"""
|
|
|
54 |
image = gr.Image(type="filepath")
|
55 |
with gr.Column():
|
56 |
model_name = gr.Dropdown(
|
57 |
+
["mistralai/Mistral-7B-Instruct-v0.1","OpenAssistant/oasst-sft-4-pythia-12b-epoch-3.5", "tiiuae/falcon-7b-instruct", "tiiuae/falcon-7b", "HuggingFaceH4/starchat-beta", "bigscience/bloom", "google/flan-t5-xxl"],
|
58 |
+
value="mistralai/Mistral-7B-Instruct-v0.1",
|
59 |
label="Choose your model!"
|
60 |
)
|
61 |
gr.Examples(["./whale.png", "./rainbow.jpeg", "./selfie.png"], inputs=image, label="Click on any example")
|