add check in function

app.py

@@ -74,6 +74,8 @@ def get_valid_hub_image_classification_model_ids():
 
 
 def predict_subset(model_id, token):
+    if token.value is None:
+            raise gr.Error("Please enter a valid token")
     valid_model_ids = get_valid_hub_image_classification_model_ids()
     if model_id not in valid_model_ids:
         raise gr.Error(
@@ -173,8 +175,6 @@ with gr.Blocks() as demo:
 
         model_id = gr.Textbox(label="model_id", value="davanstrien/autotrain-wikiart-sample2-42615108993")
         button = gr.Button("predict")
-        if token.value is None:
-            raise gr.Error("Please enter a valid token")
         gr.Markdown("## Results")
         plot = gr.BarPlot(x="labels", y="freqs", width=600, height=400, vertical=False)
         gallery = gr.Gallery()