fffiloni commited on
Commit
9b70e6e
·
verified ·
1 Parent(s): 8793418

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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 Moondream...")
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)
 
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)