Spaces:
Runtime error
Runtime error
Commit Β·
8a1dc24
1
Parent(s): 64eae51
Update app.py
Browse files
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),
|