Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -63,7 +63,7 @@ num_classes = len(list(index_to_species.keys()))
|
|
| 63 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 64 |
|
| 65 |
# Load the model
|
| 66 |
-
linear_model_name = 'linear_2025-
|
| 67 |
classify_model = models.LinearClassifier(input_dim=768, output_dim=num_classes)
|
| 68 |
classify_model.load_state_dict(torch.load(linear_model_name))
|
| 69 |
detect_model = YOLO('yolov8m_2023-10-23_best.pt')
|
|
|
|
| 63 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
| 64 |
|
| 65 |
# Load the model
|
| 66 |
+
linear_model_name = 'linear_2025-08-12.pt'
|
| 67 |
classify_model = models.LinearClassifier(input_dim=768, output_dim=num_classes)
|
| 68 |
classify_model.load_state_dict(torch.load(linear_model_name))
|
| 69 |
detect_model = YOLO('yolov8m_2023-10-23_best.pt')
|