Upload app.py with huggingface_hub

app.py

@@ -32,22 +32,17 @@ async def load_model():
             token=hf_token  # Use token for private repo access
         )
 
-        # Load with custom unpickler to handle missing modules
-        class CustomUnpickler(pickle.Unpickler):
-            def find_class(self, module, name):
-                # Redirect src.* imports to current directory
-                if module.startswith('src.'):
-                    module = module.replace('src.', '')
-                try:
-                    return super().find_class(module, name)
-                except (ModuleNotFoundError, AttributeError):
-                    # If module not found, try without the prefix
-                    return super().find_class(name, name)
-
         with open(model_path, 'rb') as f:
-            model = CustomUnpickler(f).load()
+            model_data = pickle.load(f)
+
+        # Handle both dict format and raw model
+        if isinstance(model_data, dict):
+            model = model_data['model']
+            print(f"✅ Model loaded successfully (threshold: {model_data.get('optimal_threshold', 0.5)})")
+        else:
+            model = model_data
+            print("✅ Model loaded successfully")
 
-        print("✅ Model loaded successfully")
     except Exception as e:
         print(f"❌ Error loading model: {e}")
         import traceback