sagar008 commited on
Commit
96e2541
·
verified ·
1 Parent(s): 9da1804

Update app.py

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