Update app.py

app.py

@@ -8,6 +8,20 @@ from sklearn.preprocessing import StandardScaler
 import gradio as gr
 import joblib
 
+# ✅ ประกาศ PlantSoundClassifier (เพิ่มตรงนี้)
+class PlantSoundClassifier(torch.nn.Module):
+    def __init__(self, input_dim=1024, hidden_dim=128, output_dim=2):
+        super().__init__()
+        self.fc1 = torch.nn.Linear(input_dim, hidden_dim)
+        self.relu = torch.nn.ReLU()
+        self.fc2 = torch.nn.Linear(hidden_dim, output_dim)
+    
+    def forward(self, x):
+        x = self.fc1(x)
+        x = self.relu(x)
+        x = self.fc2(x)
+        return x
+
 # 👇 โหลด processor และโมเดล HuBERT
 processor = Wav2Vec2Processor.from_pretrained("facebook/hubert-large-ls960-ft")
 model = HubertModel.from_pretrained("facebook/hubert-large-ls960-ft")
@@ -22,8 +36,8 @@ scaler = joblib.load("scaler.joblib")
 # ✅ ให้ torch รู้จัก class (safe!)
 torch.serialization.add_safe_globals({"__main__.PlantSoundClassifier": PlantSoundClassifier})
 
-# ✅ โหลด state_dict ของโมเดลอย่างปลอดภัย
-model_cls_state_dict = torch.load("classifier.pth", map_location=device, weights_only=True)
+# ✅ โหลด state_dict ของโมเดล
+model_cls_state_dict = torch.load("classifier.pth", map_location=device)
 
 # ✅ สร้าง instance classifier แล้วโหลด state_dict
 model_cls = PlantSoundClassifier().to(device)