Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -263,9 +263,9 @@ def infer(image_in, chosen_model, api_status):
|
|
| 263 |
if api_status == "api not ready yet" :
|
| 264 |
raise gr.Error("This model is not ready yet, you can pick another one instead :)")
|
| 265 |
|
| 266 |
-
gr.Info("Getting image caption with
|
| 267 |
-
|
| 268 |
-
user_prompt = get_caption_from_MD(image_in)
|
| 269 |
|
| 270 |
gr.Info("Building a musical prompt according to the image caption ...")
|
| 271 |
musical_prompt = get_musical_prompt(user_prompt, chosen_model)
|
|
|
|
| 263 |
if api_status == "api not ready yet" :
|
| 264 |
raise gr.Error("This model is not ready yet, you can pick another one instead :)")
|
| 265 |
|
| 266 |
+
gr.Info("Getting image caption with Kosmos-2...")
|
| 267 |
+
user_prompt = get_caption(image_in)
|
| 268 |
+
#user_prompt = get_caption_from_MD(image_in)
|
| 269 |
|
| 270 |
gr.Info("Building a musical prompt according to the image caption ...")
|
| 271 |
musical_prompt = get_musical_prompt(user_prompt, chosen_model)
|