Update app.py

app.py

@@ -79,8 +79,8 @@ class TextClassifier:
             num_labels=2
         ).to(self.device)
         
-        # Load your custom trained weights
-        model_path = "/home/user/.cache/model_files/model.pt"  # Adjust filename as needed
+        # Look for model file in the same directory as the code
+        model_path = "model.pt"  # Your model file should be uploaded as model.pt
         if os.path.exists(model_path):
             logger.info(f"Loading custom model from {model_path}")
             checkpoint = torch.load(model_path, map_location=self.device)