panda1835 commited on
Commit
95a8564
·
verified ·
1 Parent(s): c1c88ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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_model.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')
 
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')