Fix
Browse files
app.py
CHANGED
@@ -23,15 +23,15 @@ pad_id = tokenizer._convert_token_to_id_with_added_voc("<pad>")
|
|
23 |
|
24 |
|
25 |
def greet(input, task, lang):
|
26 |
-
if
|
27 |
model = wbmodel
|
28 |
-
elif
|
29 |
model = hgmodel
|
30 |
-
elif
|
31 |
model = ppmodel
|
32 |
-
elif
|
33 |
model = ssmodel
|
34 |
-
elif
|
35 |
model = qgmodel
|
36 |
|
37 |
|
|
|
23 |
|
24 |
|
25 |
def greet(input, task, lang):
|
26 |
+
if task == "IndicWikiBio":
|
27 |
model = wbmodel
|
28 |
+
elif task == "IndicHeadlineGeneration":
|
29 |
model = hgmodel
|
30 |
+
elif task == "IndicParaprasing":
|
31 |
model = ppmodel
|
32 |
+
elif task == "IndicSentenceSummarization":
|
33 |
model = ssmodel
|
34 |
+
elif task == "IndicQuestionGeneration":
|
35 |
model = qgmodel
|
36 |
|
37 |
|