devngho commited on
Commit
8583bb2
β€’
1 Parent(s): 977b742

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ models = {
10
  import gradio as gr
11
 
12
  def evaluate_model(input_text):
13
- return [model(input_text) for model in models.values()]
14
 
15
  # Gradio interface
16
  with gr.Blocks() as demo:
 
10
  import gradio as gr
11
 
12
  def evaluate_model(input_text):
13
+ return [model(input_text) if model_name != 'devngho/ko_edu_classifier_v2_nlpai-lab_KoE5' else model('passage: ' + input_text) for model_name, model in models.items()]
14
 
15
  # Gradio interface
16
  with gr.Blocks() as demo: