parth parekh
commited on
Commit
·
0dc96d7
1
Parent(s):
cc5ebd5
removed gargabe roberta
Browse files
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.
|
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()
|