PierreHanna commited on
Commit
8a1dc24
Β·
1 Parent(s): 64eae51

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -4
app.py CHANGED
@@ -51,7 +51,6 @@ def process(prompt, lang):
51
  print()
52
 
53
  # Embed text
54
- #embed_query = encoder_text.predict(embed_prompt["pooled_output"]) #######
55
  embed_query = get_predict(encoder_text, prompt, preprocess_model, model)
56
 
57
  faiss.normalize_L2(embed_query)
@@ -66,11 +65,8 @@ def process(prompt, lang):
66
  print("----")
67
  for i in range(len(I[0])):
68
  print(audio_names[I[0][i]], " with distance ", D[0][i])
69
- #print(" url : ", url_dict[audio_names[I[0][i]]])
70
  print(" url : ", get_url(I[0][i], audio_names, url_dict))
71
 
72
-
73
- #return [url_dict[audio_names[I[0][0]]], url_dict[audio_names[I[0][1]]], url_dict[audio_names[I[0][2]]], url_dict[audio_names[I[0][3]]], url_dict[audio_names[I[0][4]]]] #######
74
  return [get_url(I[0][0], audio_names, url_dict), get_url(I[0][1], audio_names, url_dict), get_url(I[0][2], audio_names, url_dict), get_url(I[0][3], audio_names, url_dict), get_url(I[0][4], audio_names, url_dict)] #######
75
 
76
  inputs = [gr.Textbox(label="Input", value="type your description", max_lines=2),
 
51
  print()
52
 
53
  # Embed text
 
54
  embed_query = get_predict(encoder_text, prompt, preprocess_model, model)
55
 
56
  faiss.normalize_L2(embed_query)
 
65
  print("----")
66
  for i in range(len(I[0])):
67
  print(audio_names[I[0][i]], " with distance ", D[0][i])
 
68
  print(" url : ", get_url(I[0][i], audio_names, url_dict))
69
 
 
 
70
  return [get_url(I[0][0], audio_names, url_dict), get_url(I[0][1], audio_names, url_dict), get_url(I[0][2], audio_names, url_dict), get_url(I[0][3], audio_names, url_dict), get_url(I[0][4], audio_names, url_dict)] #######
71
 
72
  inputs = [gr.Textbox(label="Input", value="type your description", max_lines=2),