auth

app.py

@@ -14,6 +14,11 @@ from upsampling import get_pipeline, upscale_image
 
 import gradio as gr
 
+from huggingface_hub import HfApi
+
+HF_TOKEN = os.getenv('HF_TOKEN')
+hf_api = HfApi(token=HF_TOKEN)
+
 # fix randomness
 th.manual_seed(0)
 np.random.seed(0)

requirements.txt

@@ -7,4 +7,5 @@ safetensors~=0.3
 sentencepiece~=0.1
 accelerate~=0.18
 bitsandbytes~=0.38
-torch~=2.0
+torch~=2.0
+huggingface_hub==0.14.1