boltuix commited on
Commit
8f67595
·
verified ·
1 Parent(s): e2a90ea

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -148,7 +148,7 @@ model = AutoModelForSequenceClassification.from_pretrained(model_name)
148
  model.eval()
149
 
150
  # 🧪 Example input
151
- text = "Turn off the fan"
152
 
153
  # ✂️ Tokenize the input
154
  inputs = tokenizer(text, return_tensors="pt")
@@ -169,8 +169,8 @@ print(f"Predicted intent: {labels[pred]} (Confidence: {probs[0][pred]:.4f})")
169
 
170
  **Output**:
171
  ```plaintext
172
- Text: Turn off the fan
173
- Predicted intent: OFF (Confidence: 0.7824)
174
  ```
175
 
176
  *Note*: Fine-tune the model for specific classification tasks to improve accuracy.
 
148
  model.eval()
149
 
150
  # 🧪 Example input
151
+ text = "Turn on the fan"
152
 
153
  # ✂️ Tokenize the input
154
  inputs = tokenizer(text, return_tensors="pt")
 
169
 
170
  **Output**:
171
  ```plaintext
172
+ Text: Turn on the fan
173
+ Predicted intent: ON (Confidence: 0.7824)
174
  ```
175
 
176
  *Note*: Fine-tune the model for specific classification tasks to improve accuracy.