Update app.py

app.py

@@ -8,10 +8,21 @@ import os
 from flask import Flask, request, jsonify
 from huggingface_hub import HfFolder
 
-# Load the Hugging Face token from the environment variable
-token = os.getenv('HUGGINGFACE_TOKEN')
+import os
+from huggingface_hub import login, HfFolder
+
+# Load the Hugging Face token from an environment variable
+token = os.getenv('HUGGINGFACE_TOKEN')  # Ensure this is set correctly
+
+if token is None:
+    print("Error: HUGGINGFACE_TOKEN is not set.")
+else:
+    print("HUGGINGFACE_TOKEN loaded successfully.")
+
+# Authenticate with Hugging Face
+login(token=token)
 HfFolder.save_token(token)
-# Load GPT-2 model and tokenizer
+
 
 # Set environment variable to avoid floating-point errors
 os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'