boltuix commited on
Commit
94730fa
·
verified ·
1 Parent(s): 7fff39f

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -312,7 +312,7 @@ To adapt BERT-Lite for custom IoT tasks (e.g., specific smart home commands):
312
  1. **Prepare Dataset**: Collect labeled data (e.g., commands with intents or masked sentences).
313
  2. **Fine-Tune with Hugging Face**:
314
  ```python
315
- import torch
316
  from transformers import BertTokenizer, BertForSequenceClassification, Trainer, TrainingArguments
317
  from datasets import Dataset
318
  import pandas as pd
 
312
  1. **Prepare Dataset**: Collect labeled data (e.g., commands with intents or masked sentences).
313
  2. **Fine-Tune with Hugging Face**:
314
  ```python
315
+ import torch
316
  from transformers import BertTokenizer, BertForSequenceClassification, Trainer, TrainingArguments
317
  from datasets import Dataset
318
  import pandas as pd