Karan0310 commited on
Commit
66c106a
·
verified ·
1 Parent(s): e4aa2e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ def get_pydantic_core_schema(request_type, handler):
19
 
20
  Request.__get_pydantic_core_schema__ = get_pydantic_core_schema
21
  #pipe = pipeline("summarization", model="sshleifer/distilbart-cnn-6-6")
22
- text_summary=pipeline(task="summarization", model=model_path, torch_dtype=torch.bfloat16)
23
 
24
 
25
  def summary(input):
 
19
 
20
  Request.__get_pydantic_core_schema__ = get_pydantic_core_schema
21
  #pipe = pipeline("summarization", model="sshleifer/distilbart-cnn-6-6")
22
+ text_summary=pipeline(task="summarization", model="sshleifer/distilbart-cnn-12-6", torch_dtype=torch.bfloat16)
23
 
24
 
25
  def summary(input):