w11wo commited on
Commit
fc28382
1 Parent(s): 4d332f0

Fix Interface Imports

Browse files
Files changed (1) hide show
  1. utils.py +14 -0
utils.py CHANGED
@@ -32,6 +32,20 @@ def text_analysis(text, pipes: list[pipeline]):
32
  ######################
33
  ##### INTERFACE ######
34
  ######################
 
 
 
 
 
 
 
 
 
 
 
 
 
 
35
  def token_classification_interface(pipe: pipeline, examples: list[str], output_label: str, title: str, desc: str):
36
  return gr.Interface(
37
  fn=partial(tagging, pipe=pipe),
 
32
  ######################
33
  ##### INTERFACE ######
34
  ######################
35
+ def text_interface(pipe: pipeline, examples: list[str], output_label: str, title: str, desc: str):
36
+ return gr.Interface(
37
+ fn=partial(cls_inference, pipe=pipe),
38
+ inputs=[
39
+ gr.Textbox(lines=5, label="Input Text"),
40
+ ],
41
+ title=title,
42
+ description=desc,
43
+ outputs=[gr.Label(label=output_label)],
44
+ examples=examples,
45
+ allow_flagging="never",
46
+ )
47
+
48
+
49
  def token_classification_interface(pipe: pipeline, examples: list[str], output_label: str, title: str, desc: str):
50
  return gr.Interface(
51
  fn=partial(tagging, pipe=pipe),