Update app.py

app.py

@@ -9,6 +9,17 @@ import torch
 import argparse
 from transformers import AutoModel, AutoTokenizer
 
+import os
+from huggingface_hub import login
+
+# Load the API token from the environment variables
+api_token = os.getenv('HUGGINGFACE_API_TOKEN')
+if not api_token:
+    raise ValueError("No Hugging Face API token found. Please set the HUGGING_FACE_API_TOKEN environment variable.")
+
+# Login to Hugging Face Hub
+login(token=api_token, add_to_git_credential=True)    
+
 # For Nvidia GPUs.
 # python web_demo_2.5.py --device cuda