prajdabre commited on
Commit
b1af7c7
·
1 Parent(s): 3b91f8a
Files changed (1) hide show
  1. app.py +5 -5
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 choice == "IndicWikiBio":
27
  model = wbmodel
28
- elif choice == "IndicHeadlineGeneration":
29
  model = hgmodel
30
- elif choice == "IndicParaprasing":
31
  model = ppmodel
32
- elif choice == "IndicSentenceSummarization":
33
  model = ssmodel
34
- elif choice == "IndicQuestionGeneration":
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