boltuix commited on
Commit
29e1be3
·
verified ·
1 Parent(s): 88c4020

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -152,7 +152,7 @@ model = AutoModelForSequenceClassification.from_pretrained(model_name)
152
  model.eval()
153
 
154
  # 🧪 Example input
155
- text = "Turn off the fan"
156
 
157
  # ✂️ Tokenize the input
158
  inputs = tokenizer(text, return_tensors="pt")
@@ -236,7 +236,7 @@ for text, answer in tests:
236
  top_scores = torch.softmax(topk.values, dim=1)[0]
237
  guesses = [(tokenizer.decode([i]).strip().lower(), float(score)) for i, score in zip(top_ids, top_scores)]
238
  results.append({
239
- "sentence": title,
240
  "expected": answer,
241
  "predictions": guesses,
242
  "pass": answer.lower() in [g[0] for g in guesses]
 
152
  model.eval()
153
 
154
  # 🧪 Example input
155
+ text = "Turn on the AC"
156
 
157
  # ✂️ Tokenize the input
158
  inputs = tokenizer(text, return_tensors="pt")
 
236
  top_scores = torch.softmax(topk.values, dim=1)[0]
237
  guesses = [(tokenizer.decode([i]).strip().lower(), float(score)) for i, score in zip(top_ids, top_scores)]
238
  results.append({
239
+ "sentence": text,
240
  "expected": answer,
241
  "predictions": guesses,
242
  "pass": answer.lower() in [g[0] for g in guesses]