Savtale commited on
Commit
c8d776e
·
verified ·
1 Parent(s): f1a644d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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 = "Some 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