Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -52,7 +52,7 @@ class YOLODroneDetector:
|
|
| 52 |
# Try alternative loading methods
|
| 53 |
try:
|
| 54 |
# Method 1: Direct torch.load
|
| 55 |
-
checkpoint = torch.load(self.weights_path, map_location='cpu')
|
| 56 |
print("π¦ Model checkpoint loaded, attempting to extract model...")
|
| 57 |
return False # Will need more complex extraction
|
| 58 |
|
|
|
|
| 52 |
# Try alternative loading methods
|
| 53 |
try:
|
| 54 |
# Method 1: Direct torch.load
|
| 55 |
+
checkpoint = torch.load(self.weights_path, map_location='cpu', weights_only=False)
|
| 56 |
print("π¦ Model checkpoint loaded, attempting to extract model...")
|
| 57 |
return False # Will need more complex extraction
|
| 58 |
|