Update app.py

app.py

@@ -1,22 +1,26 @@
 import torch
+import streamlit as st
+import numpy as np
 import torchaudio
+import torch.nn as nn
 import torch.nn.functional as F
-from model import SpeechSeparationModel  # ต้องแน่ใจว่า import ถูกต้อง
+from model import SpeechSeparationModel
 
-# โหลดโมเดลที่เทรนแล้ว
+# ตรวจสอบว่าไฟล์โมเดลอยู่ที่ไหน
 MODEL_PATH = "speech_separation_model_epoch_3.pth"
 
+# โหลดโมเดล Speech Separation
 @st.cache_resource
 def load_model():
-    model = SpeechSeparationModel()
+    model = SpeechSeparationModel()  # ตรวจสอบว่า SpeechSeparationModel ถูก import
     model.load_state_dict(torch.load(MODEL_PATH, map_location="cpu"))
     model.eval()
     return model
 
 model = load_model()
 
+# ฟังก์ชันแยกเสียงพูดโดยใช้โมเดล
 def separate_speech(audio_signal, sample_rate):
-    """ ใช้โมเดล deep learning เพื่อแยกเสียงพูดจากเสียงรบกวน """
     audio_tensor = torch.tensor(audio_signal).float().unsqueeze(0).unsqueeze(0)  # (batch, channel, length)
     with torch.no_grad():
         separated_audio = model(audio_tensor).squeeze().numpy()