Update app.py

app.py

@@ -1,13 +1,15 @@
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
-
+import os
 # Hugging Face repository details
 # MODEL_ID = "meta-llama/CodeLlama-7b-Instruct-hf"
 MODEL_ID = "meta-llama/Llama-3.1-8B"
 
 from huggingface_hub import login
-# login()
+token = os.environ.get("hf")
+
+login(token)
 def load_model():
     """Load the Hugging Face model and tokenizer."""
     try: