Update app.py

app.py

@@ -2,19 +2,21 @@ from fastapi import FastAPI
 from pydantic import BaseModel
 from transformers import pipeline, AutoTokenizer
 import uvicorn
-
+import os
 app = FastAPI()
+HF_AUTH_TOKEN=os.environ.get("HF_TOKEN")
+
 
 MODEL_NAME = "VincentMuriuki/legal-summarizer"
-summarizer = pipeline("summarization", model=MODEL_NAME)
-tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
+summarizer = pipeline("summarization", model=MODEL_NAME, token=HF_AUTH_TOKEN)
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, token=HF_AUTH_TOKEN)
 
 class SummarizeInput(BaseModel):
     text: str
 
 class ChunkInput(BaseModel):
     text: str
-    max_tokens: int = 512 
+    max_tokens: int = 512  
 
 @app.post("/summarize")
 def summarize_text(data: SummarizeInput):