Fix app.py

app.py

@@ -8,6 +8,8 @@ global default_model_name
 default_model_name = "google/vit-base-patch16-224"
 
 def predict(image, model_name):
+    if model_name == "":
+        model_name = default_model_name
     pipe = pipeline(task="image-classification", model=model_name)
     predictions = pipe(image)
     return {p["label"]: p["score"] for p in predictions}
@@ -28,8 +30,8 @@ with gr.Blocks() as demo:
         """)
     with gr.Row():
         with gr.Column(scale=1):
-            input_model = gr.Textbox(label="Enter a custom model name:", value=default_model_name, scale=1)
-           
+            #input_model = gr.Textbox(label="Enter a custom model name:", value=default_model_name, scale=1)
+            input_model = gr.Textbox(label="Enter a custom model name:", scale=1)
                 
             gr.Markdown("Upload image")
             #images_input = gr.File(file_count="multiple", file_types=["image"], label="Input images", scale=1)