using gradio textbox to string thing

app.py

@@ -43,9 +43,9 @@ def respond(
         yield response
 
 
-def train_model():
-    print(str(hf_token), flush=True)
-    os.environ["HUGGINGFACE_TOKEN"] = str(hf_token)
+def train_model(hf_token_value):
+    print(hf_token_value, flush=True)
+    os.environ["HUGGINGFACE_TOKEN"] = hf_token_value
 
     # Load dataset
     dataset = load_dataset('json', data_files={
@@ -108,7 +108,7 @@ with demo:
         train_button = gr.Button("Start Training")
         train_output = gr.Textbox(label="Training Output")
 
-        train_button.click(train_model, outputs=train_output)
+        train_button.click(train_model, inputs=hf_token, outputs=train_output)
 
 if __name__ == "__main__":
     demo.launch()