Update app.py

app.py

@@ -209,6 +209,12 @@ def get_musical_prompt(user_prompt, chosen_model):
     return cleaned_text.lstrip("\n")
 
 def infer(image_in, chosen_model):
+    if image_in == None :
+        raise gr.Error("Please provide an image input")
+
+    if chosen_model == None :
+        raise gr.Error("Please pick a model")
+    
     gr.Info("Getting image caption with Kosmos2...")
     user_prompt = get_caption(image_in)
     
@@ -264,7 +270,8 @@ with gr.Blocks(css=css) as demo:
                             "Riffusion",
                             "Mustango"
                         ],
-                        value = "MAGNet"
+                        value = None,
+                        filterable = False
                     )
                     check_status = gr.Textbox(label="API status")
                 submit_btn = gr.Button("Make music from my pic !")
@@ -279,12 +286,12 @@ with gr.Blocks(css=css) as demo:
         with gr.Column():
             gr.Examples(
                 examples = [
-                    ["examples/monalisa.png", "MAGNet"],
-                    ["examples/santa.png", "MAGNet"],
-                    ["examples/ocean_poet.jpeg", "MAGNet"],
-                    ["examples/winter_hiking.png", "MAGNet"],
-                    ["examples/teatime.jpeg", "MAGNet"],
-                    ["examples/news_experts.jpeg", "MAGNet"]
+                    ["examples/monalisa.png"],
+                    ["examples/santa.png"],
+                    ["examples/ocean_poet.jpeg"],
+                    ["examples/winter_hiking.png"],
+                    ["examples/teatime.jpeg"],
+                    ["examples/news_experts.jpeg"]
                 ],
                 fn = infer,
                 inputs = [image_in, chosen_model],