cnealex commited on
Commit
8b2faa7
·
verified ·
1 Parent(s): 42cf708

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -10
app.py CHANGED
@@ -3,13 +3,6 @@ import gradio as gr
3
  import pandas as pd
4
 
5
 
6
-
7
- def coding_v1(model, text, codetext):
8
- classifier = pipeline("zero-shot-classification", model=model)
9
- codelist = codetext.split(';')
10
- output = classifier(text, codelist, multi_label=True)
11
- return output
12
-
13
  def coding(model, text, codetext):
14
  classifier = pipeline("zero-shot-classification", model=model)
15
  codelist = codetext.split(';')
@@ -19,7 +12,7 @@ def coding(model, text, codetext):
19
  keys = output['labels']
20
  values = output['scores']
21
  my_dict = {k: v for k, v in zip(keys, values)}
22
- return my_dict
23
 
24
  def upload_code_list(file):
25
  df = pd.read_excel(file.name, sheet_name='code')
@@ -75,10 +68,10 @@ with demo:
75
  variant='primary'
76
  )
77
  with gr.Column():
78
- #result_text = gr.JSON()
79
  result_label = gr.Label(show_label=False)
 
80
 
81
- run_btn.click(coding, [select_model, comment_text, codelist_text], result_label, scroll_to_output=True)
82
 
83
  if __name__ == "__main__":
84
  demo.launch()
 
3
  import pandas as pd
4
 
5
 
 
 
 
 
 
 
 
6
  def coding(model, text, codetext):
7
  classifier = pipeline("zero-shot-classification", model=model)
8
  codelist = codetext.split(';')
 
12
  keys = output['labels']
13
  values = output['scores']
14
  my_dict = {k: v for k, v in zip(keys, values)}
15
+ return [my_dict, oytput]
16
 
17
  def upload_code_list(file):
18
  df = pd.read_excel(file.name, sheet_name='code')
 
68
  variant='primary'
69
  )
70
  with gr.Column():
 
71
  result_label = gr.Label(show_label=False)
72
+ result_text = gr.JSON()
73
 
74
+ run_btn.click(coding, [select_model, comment_text, codelist_text], [result_label, result_label], scroll_to_output=True)
75
 
76
  if __name__ == "__main__":
77
  demo.launch()