Update README.md

README.md

@@ -1,7 +1,10 @@
 ---
 license: mit
 ---
-
+import torch
+from torch import nn
+from transformers import RobertaConfig, RobertaModel, RobertaTokenizer
+import torch.optim as optim
 
 ```python
 class HelpdeskReviewModel(nn.Module):
@@ -44,6 +47,11 @@ class HelpdeskReviewModel(nn.Module):
 ```
 
 ```python
+
+tokenizer = RobertaTokenizer.from_pretrained('roberta-base')
+model = HelpdeskReviewModel()
+
+
 def make_prediction(model, tokenizer, text, max_length=512):
     # Tokenize the input text and convert to tensor
     inputs = tokenizer.encode_plus(