Update app.py

app.py

@@ -1,4 +1,5 @@
 import streamlit as st
+import os
 import numpy as np
 import re
 import tempfile
@@ -17,8 +18,9 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 from langchain.llms.huggingface_pipeline import HuggingFacePipeline
 from huggingface_hub import login
 
+hf_token = os.getenv(HF_API_TOKEN)
 # Log in with your token (optional if already logged in via CLI)
-login(token=HF_API_TOKEN)
+login(token=hf_token)
 
 # Load the model and tokenizer
 model_name = "meta-llama/Llama-3.2-1B-Instruct"