Updated FastAPI code

config.py

@@ -1,22 +1,25 @@
 import psutil
 import os
-from dotenv import load_dotenv
 import nltk
 
-
-load_dotenv()
+# Whisper settings
 WHISPER_MODEL_NAME = "tiny"
 WHISPER_DEVICE = "cpu"
 WHISPER_COMPUTE_TYPE = "int8"
-PYANNOTE_AUTH_TOKEN = os.getenv("HUGGINGFACE_API_KEY")
 
-#SUMMARIZER_MODEL = "google/flan-t5-base"
+# Use token from Hugging Face Space secrets
+PYANNOTE_AUTH_TOKEN = os.environ.get("HUGGINGFACE_API_KEY")
+
+# Summarization settings
+# SUMMARIZER_MODEL = "google/flan-t5-base"
 SUMMARIZER_MODEL = "facebook/bart-large-cnn"
 SUMMARIZER_MAX_LENGTH = 150
 SUMMARIZER_MIN_LENGTH = 50
 
+# NLP settings
 SPACY_MODEL = "en_core_web_sm"
 CPU_THREADS = max(1, psutil.cpu_count(logical=False))
 
+# Check that token is set
 if not PYANNOTE_AUTH_TOKEN:
-    raise ValueError("HUGGINGFACE_API_KEY not set in environment variables")
+    raise ValueError("HUGGINGFACE_API_KEY is not set in environment variables")

requirements.txt

@@ -1,6 +1,5 @@
 fastapi
 uvicorn
-python-dotenv
 transformers
 torch
 faster_whisper