Update app.py

app.py

@@ -28,12 +28,12 @@ def pool_span_scores(score_dicts, sent_lens):
                                     for token_id in range(sent_lens[sent_id])]
                         for sent_id in range(len(sent_lens))]
                         
-    r = [[TAGS[ps] for ps in sent_ps]for sent_ps in pooled_scores]
+    r = [[TAGS[ps] for ps in sent_ps] for sent_ps in pooled_scores]
     return r 
 
 def predict_label(text):
     model_path = 'models/span_model'
-    ip = input_text.split()
+    ip = text.split()
     ip_len = [length(ip)]
     scores = extract_spannet_scores(model_path,ip,ip_len, pos_col=1, task_col=2)
     pooled_scores = pool_span_scores(scores, ip_len)