prajdabre commited on
Commit
f60d71b
·
1 Parent(s): 19cbaa3
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -9,7 +9,7 @@ tokenizer = AlbertTokenizer.from_pretrained("ai4bharat/MultiIndicWikiBioSS", do_
9
 
10
  # Or use tokenizer = AlbertTokenizer.from_pretrained("ai4bharat/IndicBART-XLSum", do_lower_case=False, use_fast=False, keep_accents=True)
11
 
12
- xlsummodel = AutoModelForSeq2SeqLM.from_pretrained("ai4bharat/IndicBART-XLSum")
13
  qgmodel = AutoModelForSeq2SeqLM.from_pretrained("ai4bharat/MultiIndicQuestionGenerationSS").eval()
14
  hgmodel = AutoModelForSeq2SeqLM.from_pretrained("ai4bharat/MultiIndicHeadlineGenerationSS").eval()
15
  ssmodel = AutoModelForSeq2SeqLM.from_pretrained("ai4bharat/MultiIndicSentenceSummarizationSS").eval()
@@ -29,7 +29,7 @@ def greet(input, task, lang):
29
  model = wbmodel
30
  elif task == "IndicHeadlineGeneration":
31
  model = hgmodel
32
- elif task == "IndicParaprasing":
33
  model = ppmodel
34
  elif task == "IndicSentenceSummarization":
35
  model = ssmodel
 
9
 
10
  # Or use tokenizer = AlbertTokenizer.from_pretrained("ai4bharat/IndicBART-XLSum", do_lower_case=False, use_fast=False, keep_accents=True)
11
 
12
+ # xlsummodel = AutoModelForSeq2SeqLM.from_pretrained("ai4bharat/IndicBART-XLSum")
13
  qgmodel = AutoModelForSeq2SeqLM.from_pretrained("ai4bharat/MultiIndicQuestionGenerationSS").eval()
14
  hgmodel = AutoModelForSeq2SeqLM.from_pretrained("ai4bharat/MultiIndicHeadlineGenerationSS").eval()
15
  ssmodel = AutoModelForSeq2SeqLM.from_pretrained("ai4bharat/MultiIndicSentenceSummarizationSS").eval()
 
29
  model = wbmodel
30
  elif task == "IndicHeadlineGeneration":
31
  model = hgmodel
32
+ elif task == "IndicParaphrasing":
33
  model = ppmodel
34
  elif task == "IndicSentenceSummarization":
35
  model = ssmodel