Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import AutoModelForCausalLM, AutoTokenizer, Trainer, TrainingArguments
 from datasets import load_dataset
+from huggingface_hub import login
 import os
 
 # Retrieve the Hugging Face token from the Space secrets