Update app.py

app.py

@@ -231,7 +231,7 @@ def cosine_similarity_numpy(src_array, trg_array):
 
 def select_best_output(outputs, embeddings, src_embeddings, top_k=10):
 
-    emb_sims = cosine_similarity_numpy(np.array(src_embeddings), np.array(embeddings))
+    emb_sims = cosine_similarity_numpy(src_embeddings, embeddings)
 
     sorted_emb_sims = sorted(emb_sims, reverse=True)
 
@@ -273,9 +273,9 @@ def Classify_MIDI_Genre(input_midi):
 
     src_score = load_midi(input_midi.name)
 
-    inp = torch.LongTensor([src_score]).to(device_type)
+    inp = torch.LongTensor([src_score]).cuda()
 
-    src_emb = get_embeddings(inp).tolist()
+    src_emb = get_embeddings(inp)
 
     print('Done!')
 
@@ -290,7 +290,7 @@ def Classify_MIDI_Genre(input_midi):
 
     #===============================================================================
 
-    model.to(device_type)
+    model.cuda()
     model.eval()
 
     #===============================================================================