Spaces:

d0r1h
/

Hindi_News_Summarizer

Sleeping

d0r1h commited on May 28, 2022

Commit

d7989d6

1 Parent(s): 32027d2

Updated bart model

Files changed (1) hide show

summarizer.py CHANGED Viewed

@@ -12,7 +12,7 @@ def summarize(input_, model):
   if model == "T5":
     checkpoint = "csebuetnlp/mT5_multilingual_XLSum"
   elif model == "BART":
-    checkpoint = "ai4bharat/IndicBART"
   WHITESPACE_HANDLER = lambda k: re.sub('\s+', ' ', re.sub('\n+', ' ', k.strip()))

   if model == "T5":
     checkpoint = "csebuetnlp/mT5_multilingual_XLSum"
   elif model == "BART":
+    checkpoint = "ai4bharat/IndicBART-XLSum"
   WHITESPACE_HANDLER = lambda k: re.sub('\s+', ' ', re.sub('\n+', ' ', k.strip()))