Update app.py

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer
+from huggingface_hub import login  # Import login function
 import torch
+import os
 import re
 
 login(token=os.getenv("HF_TOKEN"))