Update app.py

app.py

@@ -1,6 +1,11 @@
 import streamlit as st
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
+from huggingface_hub import login
+import os
+
+login(token=os.getenv("HF_TOKEN"))
+
 
 # Load the model and tokenizer
 MODEL_NAME = "google/gemma-2b-it"