Update app.py

app.py

@@ -2,6 +2,7 @@ import joblib
 import gradio as gr
 from datasets import Dataset, DatasetDict, load_dataset
 from huggingface_hub import login
+import os
 
 token = os.getenv('HF_TOKEN')
 login(token, add_to_git_credential=True,write_permission=True )