Add application file

app.py

@@ -89,18 +89,15 @@ def get_embedding(data, embedder_net, device, n_threshold=-1):
     
 model_path = "./speech_id_checkpoint/saved_02.model"
 
-
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-
-embedder_net = SpeechEmbedder().to(device)
-embedder_net.load_state_dict(torch.load(model_path, map_location=device))
+embedder_net = SpeechEmbedder()
+embedder_net.load_state_dict(torch.load(model_path))
 embedder_net.eval()
 
 def process_audio(audio1, audio2, threshold):
-    e1 = get_embedding(audio1, embedder_net, device)
+    e1 = get_embedding(audio1, embedder_net, torch.device("cpu"))
     if(e1 is None):
         return "No Voice Detected in file 1"
-    e2 = get_embedding(audio2, embedder_net, device)
+    e2 = get_embedding(audio2, embedder_net, torch.device("cpu"))
     if(e2 is None):
         return "No Voice Detected in file 2"