parth parekh commited on
Commit
0dc96d7
·
1 Parent(s): cc5ebd5

removed gargabe roberta

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ class ContactDetector:
27
  probabilities = softmax(outputs.logits, dim=1)
28
  return probabilities[0][1].item() # Probability of contact info
29
 
30
- def is_contact_info(self, text, threshold=0.6):
31
  return self.detect_contact_info(text) > threshold
32
 
33
  detector = ContactDetector()
 
27
  probabilities = softmax(outputs.logits, dim=1)
28
  return probabilities[0][1].item() # Probability of contact info
29
 
30
+ def is_contact_info(self, text, threshold=0.5):
31
  return self.detect_contact_info(text) > threshold
32
 
33
  detector = ContactDetector()