Fix
Browse files
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 == "
|
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
|