non2013 commited on
Commit
dbda7cf
·
1 Parent(s): f227fd1

update interface

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -58,8 +58,8 @@ def classify_question(text):
58
  # Create a list of probabilities for each model
59
  probs = {
60
  "Probability": float(avg_pred),
61
- "Model Probabilities": {"Model 1": float(pred1), "Model 2": float(pred2), "Model 3": float(pred3), "Model 4": float(pred4), "visible": False},
62
- "Sequence": {"value": seq, "visible": False}
63
  }
64
 
65
  return label, probs
 
58
  # Create a list of probabilities for each model
59
  probs = {
60
  "Probability": float(avg_pred),
61
+ "Model Probabilities": {"Model 1": float(pred1), "Model 2": float(pred2), "Model 3": float(pred3), "Model 4": float(pred4)},
62
+ "Sequence": seq
63
  }
64
 
65
  return label, probs