Lord-Raven commited on
Commit
259344b
·
1 Parent(s): 81bb8c3

Trying to fix parameter issue.

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,7 +14,8 @@ classifier = pipeline(
14
  model=model,
15
  tokenizer=tokenizer,
16
  top_k=None,
17
- function_to_apply="sigmoid", # optional as is the default for the task
 
18
  )
19
 
20
  def predict(param_0):
@@ -26,5 +27,4 @@ demo = gr.Interface(
26
  outputs = 'text',
27
  )
28
 
29
- # demo = gr.Interface.from_pipeline(classifier)
30
  demo.launch()
 
14
  model=model,
15
  tokenizer=tokenizer,
16
  top_k=None,
17
+ function_to_apply="sigmoid",
18
+ binary_output=True,
19
  )
20
 
21
  def predict(param_0):
 
27
  outputs = 'text',
28
  )
29
 
 
30
  demo.launch()