Update app.py

app.py

@@ -28,7 +28,8 @@ model_hubert = HubertModel.from_pretrained(model_name)
 model_hubert.eval()
 
 # 3️⃣ โหลด classifier ที่ fine-tune แล้ว
-state_dict = torch.load("classifier.pt", map_location="cpu")
+model_cls = PlantSoundClassifier()
+state_dict = torch.load("classifier.pt", map_location="cpu",weights_only=False )
 model_cls.eval()
 
 # 4️⃣ โหลด scaler