cheesexuebao commited on
Commit
8185892
1 Parent(s): cf5b520

fix: ✨ 多标签变成多分类

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -29,7 +29,7 @@ def single_sentence(sentence):
29
  dct = manager[model_name]
30
  model, tokenizer = dct['model'], dct['tokenizer']
31
  predictions = predict_single(sentence, tokenizer, model, device)
32
- df.append([model_name] + predictions)
33
  return df
34
 
35
  def csv_process(csv_file, attr="content"):
@@ -73,7 +73,7 @@ with gr.Blocks(theme=my_theme, title='Murphy') as demo:
73
  tbox_input = gr.Textbox(label="Input",
74
  info="Please input a sentence here:")
75
  gr.Markdown("""
76
- # Detailed Information About our Model
77
  ...
78
  """)
79
  tab_output = gr.DataFrame(label='Probability Predictions:',
 
29
  dct = manager[model_name]
30
  model, tokenizer = dct['model'], dct['tokenizer']
31
  predictions = predict_single(sentence, tokenizer, model, device)
32
+ df.append(predictions)
33
  return df
34
 
35
  def csv_process(csv_file, attr="content"):
 
73
  tbox_input = gr.Textbox(label="Input",
74
  info="Please input a sentence here:")
75
  gr.Markdown("""
76
+ # Detailed information about our model:
77
  ...
78
  """)
79
  tab_output = gr.DataFrame(label='Probability Predictions:',