Hugging Face's logo

Spaces:
PageOrg
/
summaryapi
Runtime error

App Files Community
quyip commited on
Commit
5513f06
·
1 Parent(s): 8025845

fix

Browse files
Files changed (1) hide show
  1. utils/summary_utils.py +1 -1
utils/summary_utils.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline
4
  AiSummaryVersion = 1
5
  summarization_pipeline = pipeline("summarization", model="csebuetnlp/mT5_multilingual_XLSum", max_length=512, min_length=50)
6
  en_translation_pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-mul-en")
7
- text_to_tags_pipe = pipeline('text2text-generation', model='PageOrg/t5-small-tagging-text')
8
 
9
 
10
  def summarize(id: str, text: str):
 
4
  AiSummaryVersion = 1
5
  summarization_pipeline = pipeline("summarization", model="csebuetnlp/mT5_multilingual_XLSum", max_length=512, min_length=50)
6
  en_translation_pipe = pipeline("translation", model="Helsinki-NLP/opus-mt-mul-en")
7
+ text_to_tags_pipe = pipeline('text2text-generation', model='PageOrg/t5-small-tagging-text', max_length=64)
8
 
9
 
10
  def summarize(id: str, text: str):