huggingface login?

app.py

@@ -4,11 +4,15 @@ import uuid
 import gradio as gr
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import torch
+from huggingface_hub import login
+
+# Authenticate with Hugging Face
+login()
 
 # Load model and tokenizer
 model_name = "meta-llama/Meta-Llama-3.1-8B-Instruct"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto")
+tokenizer = AutoTokenizer.from_pretrained(model_name, use_auth_token=True)
+model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.float16, device_map="auto", use_auth_token=True)
 
 def chat_with_model(messages):
     input_ids = tokenizer.encode(str(messages), return_tensors="pt").to(model.device)
@@ -41,4 +45,4 @@ def main():
     demo.launch()
 
 if __name__ == "__main__":
-    main()
+    main()

requirements.txt

@@ -1,2 +1,3 @@
 transformers
-torch
+torch
+huggingface_hub