hotfix

dayyass · Jul 18, 2022 · 6ffa055 · 6ffa055
1 parent 94e3860
commit 6ffa055
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/qaner/inference_utils.py b/qaner/inference_utils.py
@@ -86,7 +86,7 @@ def get_top_valid_spans(
                 span = Span(
                     token=context[start_context_char_char:end_context_char_char],
                     label=inv_prompt_mapper[  # TODO: add inference exception
-                        question_list[i].lstrip("What is the ").rstrip("?")
+                        question_list[i].split(r"What is the ")[-1].rstrip(r"?")
                     ],
                     start_context_char_pos=start_context_char_char,
                     end_context_char_pos=end_context_char_char,