Update app.py

app.py

@@ -2,9 +2,13 @@ import gc
 import torch
 import streamlit as st
 from transformers import AutoTokenizer, AutoModelForCausalLM
+from huggingface_hub import login
 import re
 import os
 
+HF_TOKEN = os.getenv("HF_TOKEN")
+login(token=HF_TOKEN)
+
 MODELS = {
     "athena-1": {
         "name": "🦁 Athena-Flash",