fixed model not going to correct device

inference.py

@@ -32,6 +32,7 @@ class DistilBERTSent(nn.Module):
 def initialize(path="models/model.pt"):
     model = DistilBERTSent()
     model.load_state_dict(torch.load(path, map_location=device))
+    model.to(device)
     model.eval()
     return model