Update app.py

app.py

@@ -208,13 +208,15 @@ def get_musical_prompt(user_prompt, chosen_model):
     print(f"SUGGESTED Musical prompt: {cleaned_text}")
     return cleaned_text.lstrip("\n")
 
-def infer(image_in, chosen_model):
+def infer(image_in, chosen_model, api_status):
     if image_in == None :
         raise gr.Error("Please provide an image input")
 
-    print(f"CHOSEN MODEL:  {chosen_model}")
     if chosen_model == [] :
         raise gr.Error("Please pick a model")
+
+    if api_status == "api not ready yet" :
+        raise gr.Error("This model is not ready yet, you can pick another one instead :)")
     
     gr.Info("Getting image caption with Kosmos2...")
     user_prompt = get_caption(image_in)
@@ -337,7 +339,8 @@ with gr.Blocks(css=css) as demo:
         fn = infer,
         inputs = [
             image_in,
-            chosen_model
+            chosen_model,
+            check_status
         ],
         outputs =[
             caption,