Testing model location

model.py

@@ -40,7 +40,7 @@ def load_models():
     model.lm_head = nn.Identity()
     tokenizer = RobertaTokenizerFast.from_pretrained("roberta-base")
     my_classifier = SimpleClassifier(768, 768, 1)
-    weights_path = os.path.join(__file__, "..", "twitter_model_91_5-.pth")
+    weights_path = "twitter_model_91_5-.pth"
     my_classifier.load_state_dict(torch.load(weights_path, map_location=device))
     my_classifier.eval()
     return {