sagar008 commited on
Commit
2609cab
·
verified ·
1 Parent(s): 4828140

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,8 +8,8 @@ HF_AUTH_TOKEN=os.environ.get("HF_TOKEN")
8
 
9
 
10
  MODEL_NAME = "VincentMuriuki/legal-summarizer"
11
- summarizer = pipeline("summarization", model=MODEL_NAME, token=HF_AUTH_TOKEN)
12
- tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, token=HF_AUTH_TOKEN)
13
 
14
  class SummarizeInput(BaseModel):
15
  text: str
 
8
 
9
 
10
  MODEL_NAME = "VincentMuriuki/legal-summarizer"
11
+ summarizer = pipeline("summarization", model=MODEL_NAME, use_auth_token=HF_AUTH_TOKEN)
12
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, use_auth_token=HF_AUTH_TOKEN)
13
 
14
  class SummarizeInput(BaseModel):
15
  text: str