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 = '
|
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-07-18.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')
|