Spaces:
Sleeping
Sleeping
from transformers import pipeline | |
import os | |
def load_summarization_model(): | |
"""Loads the summarization model. Check for HUGGINGFACE_API_TOKEN first.""" | |
api_token = os.environ.get("HUGGINGFACE_API_TOKEN") | |
model_name = "facebook/bart-large-cnn" # Or whatever | |
if not api_token: | |
print("HUGGINGFACE_API_TOKEN not found. Summarization will not work.") | |
return None | |
try: | |
summarizer = pipeline("summarization", model=model_name, token=api_token) | |
print(f"Summarization Model {model_name} Loaded...") | |
return summarizer | |
except Exception as e: | |
print(f"Model load error: {e}") | |
return None |