Update app.py

app.py

@@ -10,6 +10,9 @@ import faiss
 import csv
 import datetime
 
+from huggingface_hub import hf_hub_download
+file_path = hf_hub_download("PierreHanna/TextRetrieval", "encoder_text_retrievaltext_bmg_221022_54.h5",
+                                use_auth_token=os.environ['TOKEN'])
 
 # NO GPU
 os.environ['CUDA_VISIBLE_DEVICES'] = '-1'
@@ -70,7 +73,8 @@ def process(prompt, lang):
 
     # Embed text
     #from models import *
-    encoder_text = tf.keras.models.load_model('encoder_text_retrievaltext_bmg_221022_54.h5')
+    #encoder_text = tf.keras.models.load_model('encoder_text_retrievaltext_bmg_221022_54.h5')
+    encoder_text = tf.keras.models.load_model(file_path)
     embed_query  = encoder_text.predict(embed_prompt["pooled_output"])
     faiss.normalize_L2(embed_query)
     print("   text embed computed.")