iimran commited on
Commit
7e57720
·
verified ·
1 Parent(s): 3657c27

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -78,8 +78,8 @@ if __name__ == "__main__":
78
  for text in example_texts:
79
  result = pipeline.predict(text)
80
  print(f"Input: {text}")
81
- print(f"Prediction: {result['label']} ({result['confidence']:.2%})")
82
- print(f"Probabilities: Inappropriate={result['probabilities'][0]:.2%}, Not Inappropriate={result['probabilities'][1]:.2%}")
83
  print("-" * 80)
84
 
85
  # Define a function for Gradio to use
 
78
  for text in example_texts:
79
  result = pipeline.predict(text)
80
  print(f"Input: {text}")
81
+ print(f"Prediction: {result['label']} ")
82
+ #print(f"Probabilities: Inappropriate={result['probabilities'][0]:.2%}, Not Inappropriate={result['probabilities'][1]:.2%}")
83
  print("-" * 80)
84
 
85
  # Define a function for Gradio to use