dmariko commited on
Commit
40825c7
1 Parent(s): 79f1e27

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -59,7 +59,7 @@ if __name__ == '__main__':
59
  #Create a gradio app with a button that calls predict()
60
  app = gr.Interface(
61
  fn=predict,
62
- inputs=[gr.inputs.Dropdown(model_name_list, label="Model Name"), 'text'], outputs=['label'],
63
  examples = [[MODEL_BUF["name"], text]],
64
  title="FReE",
65
  description="Financial relations classifier"
 
59
  #Create a gradio app with a button that calls predict()
60
  app = gr.Interface(
61
  fn=predict,
62
+ inputs=[gr.inputs.Dropdown(model_name_list, label="Model Name"), 'text'], outputs=['Output'],
63
  examples = [[MODEL_BUF["name"], text]],
64
  title="FReE",
65
  description="Financial relations classifier"