sohomghosh
/

finrad_model

Model card Files Files and versions Community

sohomghosh commited on Jun 27, 2022

Commit

0b585fd

•

1 Parent(s): 17ef62c

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -6,7 +6,7 @@ How to load the model and generate predictions?
 ```python
 import torch
 import transformers
-from torch.utils.data import Dataset, dataLoader
 from transformers import RobertaModel, RobertaTokenizer, BertModel, BertTokenizer
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

 ```python
 import torch
 import transformers
+from torch.utils.data import Dataset, DataLoader
 from transformers import RobertaModel, RobertaTokenizer, BertModel, BertTokenizer
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")