Update app.py

app.py

@@ -9,7 +9,7 @@ import matplotlib.pyplot as plt
 
 import io
 import requests
-from huggingface_hub import notebook_login
+from huggingface_hub import login
 
 import os
 
@@ -23,9 +23,9 @@ os.system('pip install accelerate')
 os.system('pip install transformers[sentencepiece]')
 os.system('pip install Pillow')
 os.system('pip install gradio')
+AUTH_TOKEN = os.environ.getattribute("AUTH_TOKEN")
 
-
-notebook_login()
+login(token=AUTH_TOKEN)