use login button

text_classification.py

@@ -102,9 +102,8 @@ def hf_inference_api(model_id, hf_token, payload):
     except Exception:
         return {"error": response.content}
     
-def preload_hf_inference_api(model_id):
+def preload_hf_inference_api(model_id, hf_token):
     payload = {"inputs": "This is a test", "options": {"use_cache": True, }}
-    hf_token = os.environ.get(HF_WRITE_TOKEN, default="")
     hf_inference_api(model_id, hf_token, payload)
 
 def check_model_pipeline(model_id):

text_classification_ui_helpers.py

@@ -179,11 +179,11 @@ def list_labels_and_features_from_dataset(ds_labels, ds_features, model_labels,
 
 
 def precheck_model_ds_enable_example_btn(
-    model_id, dataset_id, dataset_config, dataset_split
+    model_id, dataset_id, dataset_config, dataset_split, profile: gr.OAuthProfile | None, oath_token: gr.OAuthToken | None
 ):  
     model_id = strip_model_id_from_url(model_id)
     model_task = check_model_task(model_id)
-    preload_hf_inference_api(model_id)
+    preload_hf_inference_api(model_id, oath_token.token)
 
     if dataset_config is None or dataset_split is None or len(dataset_config) == 0:
         return (