specified token on HF

app.py

@@ -1,6 +1,8 @@
 from transformers import pipeline
 import torch
 import gradio as gr
+import os
+from huggingface_hub import login
 from utils import parse_args
 
 
@@ -9,6 +11,13 @@ args = parse_args()
 model = args['model']
 model_name = "clinical-assistance/" + model
 
+# Load secrets
+token = os.getenv("TOKEN")
+assert token is not None, 'Hugging Face token has not been specified. Please specify your token in a .env file.'
+
+# Login to Hugging Face
+login(token=token)
+
 # Init model
 device = 'cuda:0' if torch.cuda.is_available() else 'cpu'
 generate_kwargs = {"language":"<|es|>", "task": "transcribe"}