Spaces:
Running
on
Zero
Running
on
Zero
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)
|