PierreHanna commited on
Commit
0d3a598
·
verified ·
1 Parent(s): 7803504

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -15
app.py CHANGED
@@ -36,21 +36,21 @@ def process(prompt, lang):
36
  print()
37
 
38
  #try :
39
- embed_query = get_predict(encoder_text, prompt, preprocess_model, model)
40
- do_normalize(embed_query)
41
- D, I = get_distance(index, embed_query, TOP)
42
- print(I)
43
- print(D)
44
- print("----")
45
- for i in range(len(I[0])):
46
- print(audio_names[I[0][i]], " with distance ", D[0][i])
47
- print(" url : ", get_url_myma(I[0][i], audio_names, url_dict))
48
-
49
- return [audio_names[I[0][0]].split('.')[0], get_url_myma(I[0][0], audio_names, url_dict),
50
- audio_names[I[0][1]].split('.')[0], get_url_myma(I[0][1], audio_names, url_dict),
51
- audio_names[I[0][2]].split('.')[0], get_url_myma(I[0][2], audio_names, url_dict),
52
- audio_names[I[0][3]].split('.')[0], get_url_myma(I[0][3], audio_names, url_dict),
53
- audio_names[I[0][4]].split('.')[0], get_url_myma(I[0][4], audio_names, url_dict)]
54
  '''
55
  except:
56
  return ["Error input - please try again", "",
 
36
  print()
37
 
38
  #try :
39
+ embed_query = get_predict(encoder_text, prompt, preprocess_model, model)
40
+ do_normalize(embed_query)
41
+ D, I = get_distance(index, embed_query, TOP)
42
+ print(I)
43
+ print(D)
44
+ print("----")
45
+ for i in range(len(I[0])):
46
+ print(audio_names[I[0][i]], " with distance ", D[0][i])
47
+ print(" url : ", get_url_myma(I[0][i], audio_names, url_dict))
48
+
49
+ return [audio_names[I[0][0]].split('.')[0], get_url_myma(I[0][0], audio_names, url_dict),
50
+ audio_names[I[0][1]].split('.')[0], get_url_myma(I[0][1], audio_names, url_dict),
51
+ audio_names[I[0][2]].split('.')[0], get_url_myma(I[0][2], audio_names, url_dict),
52
+ audio_names[I[0][3]].split('.')[0], get_url_myma(I[0][3], audio_names, url_dict),
53
+ audio_names[I[0][4]].split('.')[0], get_url_myma(I[0][4], audio_names, url_dict)]
54
  '''
55
  except:
56
  return ["Error input - please try again", "",