zhangfan
commited on
Commit
•
1ac96d4
1
Parent(s):
e87e483
Update README.md
Browse files
README.md
CHANGED
@@ -58,7 +58,7 @@ model = AutoModelForSequenceClassification.from_pretrained("Fan-s/reddit-tc-bert
|
|
58 |
post = "don't make gravy with asbestos."
|
59 |
response = "i'd expect someone with a culinary background to know that. since we're talking about school dinner ladies, they need to learn this pronto."
|
60 |
|
61 |
-
# Predict whether the two sentences
|
62 |
def predict(post, response, max_seq_length=128):
|
63 |
with torch.no_grad():
|
64 |
args = (post, response)
|
|
|
58 |
post = "don't make gravy with asbestos."
|
59 |
response = "i'd expect someone with a culinary background to know that. since we're talking about school dinner ladies, they need to learn this pronto."
|
60 |
|
61 |
+
# Predict whether the two sentences are matched
|
62 |
def predict(post, response, max_seq_length=128):
|
63 |
with torch.no_grad():
|
64 |
args = (post, response)
|