Update inference.py

inference.py

@@ -5,7 +5,6 @@ from typing import List, Dict, Set
 from pathlib import Path
 from transformers import DistilBertTokenizer, DistilBertModel
 
-
 class Nnet(nn.Module):
     def __init__(self) -> None:
         super().__init__()
@@ -78,4 +77,6 @@ class InferenceModel:
         print(probs)
         # map classes to lables
         return self.get_lables(topp_calsses)
+
+# restart