sagar008 commited on
Commit
5034014
·
verified ·
1 Parent(s): 9e6e587

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,8 +16,8 @@ app = FastAPI()
16
  HF_AUTH_TOKEN = os.getenv("HF_TOKEN")
17
 
18
  MODEL_NAME = "VincentMuriuki/legal-summarizer"
19
- summarizer = pipeline("summarization", model=MODEL_NAME, use_auth_token=HF_AUTH_TOKEN)
20
- tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, use_auth_token=HF_AUTH_TOKEN)
21
 
22
  class SummarizeInput(BaseModel):
23
  text: str
 
16
  HF_AUTH_TOKEN = os.getenv("HF_TOKEN")
17
 
18
  MODEL_NAME = "VincentMuriuki/legal-summarizer"
19
+ summarizer = pipeline("summarization", model=MODEL_NAME, token=HF_AUTH_TOKEN)
20
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, token=HF_AUTH_TOKEN)
21
 
22
  class SummarizeInput(BaseModel):
23
  text: str