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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -7
app.py CHANGED
@@ -35,29 +35,30 @@ def process(prompt, lang):
35
  print('*************')
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
  except:
55
  return ["Error input - please try again", "",
56
  "Error input - please try again", "",
57
  "Error input - please try again", "",
58
  "Error input - please try again", "",
59
  "Error input - please try again", "",]
60
-
61
  inputs = [gr.Textbox(label="Input", value="type your description", max_lines=2),
62
  gr.Radio(label="Language", choices=["en"], value="en")]
63
 
 
35
  print('*************')
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", "",
57
  "Error input - please try again", "",
58
  "Error input - please try again", "",
59
  "Error input - please try again", "",
60
  "Error input - please try again", "",]
61
+ '''
62
  inputs = [gr.Textbox(label="Input", value="type your description", max_lines=2),
63
  gr.Radio(label="Language", choices=["en"], value="en")]
64