Update app.py

app.py

@@ -17,6 +17,20 @@
 from argparse import Namespace
 args = Namespace()
 
+import os
+from transformers import AutoTokenizer, AutoModel
+
+# Step 1: Get Hugging Face token from the secret
+hf_token = os.environ.get("llmtoken")  # HF_TOKEN is the name of the secret in your Space
+
+# Step 2: Specify the model
+model_name = "meta-llama/Llama-3.2-3B"
+
+# Step 3: Load tokenizer and model using the token
+tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=hf_token)
+model = AutoModel.from_pretrained(model_name, use_auth_token=hf_token)
+
+
 arg_dict = {
     'run_gradio': True, 
     'demo_public': False,