Update README.md
Browse files
README.md
CHANGED
@@ -25,7 +25,7 @@ model = AutoModelForSequenceClassification.from_pretrained("Savtale/User-Intenti
|
|
25 |
tokenizer = AutoTokenizer.from_pretrained("Savtale/User-Intention-Recognition")
|
26 |
|
27 |
# User is talking with a chatbot
|
28 |
-
input_user_text = "
|
29 |
|
30 |
# Tokenizing text
|
31 |
inputs = tokenizer(input_user_text.lower(), return_tensors="pt")
|
@@ -40,7 +40,7 @@ probabilities = torch.softmax(logits, dim=-1) # Top Match
|
|
40 |
predicted_class = int(torch.argmax(probabilities)) # Class no.
|
41 |
str_predicted_class = class_dict[str(predicted_class)] # Class String
|
42 |
|
43 |
-
print(f"Predicted User Intention: {str_predicted_class}")
|
44 |
|
45 |
```
|
46 |
|
|
|
25 |
tokenizer = AutoTokenizer.from_pretrained("Savtale/User-Intention-Recognition")
|
26 |
|
27 |
# User is talking with a chatbot
|
28 |
+
input_user_text = "Sounds good. I'm interested in trying the free trial. How do I sign up?"
|
29 |
|
30 |
# Tokenizing text
|
31 |
inputs = tokenizer(input_user_text.lower(), return_tensors="pt")
|
|
|
40 |
predicted_class = int(torch.argmax(probabilities)) # Class no.
|
41 |
str_predicted_class = class_dict[str(predicted_class)] # Class String
|
42 |
|
43 |
+
print(f"Predicted User Intention: {str_predicted_class}") # Predicted User Intention: Request a Demo
|
44 |
|
45 |
```
|
46 |
|