Dawid Motyka commited on
Commit
24fa0b5
·
1 Parent(s): f5d4fa9

minor interface changes

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -45,14 +45,14 @@ pipeline = StancePipeline(model=model, tokenizer=tokenizer, batch_size=1)
45
 
46
  def predict(text, target):
47
  predictions = pipeline({'text': text, 'target': target})
48
- return predictions['stance'], predictions['score']
49
 
50
 
51
  gradio_app = gr.Interface(
52
  predict,
53
- inputs=[gr.TextArea(label="Text", placeholder="text"),
54
- gr.Textbox(label="Target", placeholder="stance target")],
55
- outputs=[gr.Label(label="Stance"), gr.Label(label="Score")],
56
  title="Polish stance detection",
57
  )
58
 
 
45
 
46
  def predict(text, target):
47
  predictions = pipeline({'text': text, 'target': target})
48
+ return f'{predictions["stance"]} ({predictions["score"]:.3f})'
49
 
50
 
51
  gradio_app = gr.Interface(
52
  predict,
53
+ inputs=[gr.TextArea(label="Text", placeholder="Enter text here..."),
54
+ gr.Textbox(label="Target", placeholder="Enter stance target here...")],
55
+ outputs=[gr.Label(label="Stance class")],
56
  title="Polish stance detection",
57
  )
58