Update app.py

app.py

@@ -22,11 +22,11 @@ os.environ['TF_CPP_MIN_LOG_LEVEL'] = '3'
 
 
 python_path = hf_hub_download(repo_id="PierreHanna/TextRetrieval", repo_type="space", filename="models.py",
-                            use_auth_token=os.environ['TOKEN'])
-from models import *
+                            use_auth_token=os.environ['TOKEN'], subfolder="ph")
+from ph.models import *
 
 
-def make_bert_preprocess_model(sentence_features, tfhub_handle_preprocess, seq_length=128):
+def make_preprocess_model(sentence_features, tfhub_handle_preprocess, seq_length=128):
   """Returns Model mapping string features to BERT inputs.
   """
 
@@ -60,7 +60,7 @@ def process(prompt, lang):
     TOP = 10
 
 
-    bert_preprocess_model = make_bert_preprocess_model(['my_input'], tfhub_handle_preprocess, seq_length = MAX_LENGTH)
+    bert_preprocess_model = make_preprocess_model(['my_input'], tfhub_handle_preprocess, seq_length = MAX_LENGTH)
     bert_model = hub.KerasLayer(tfhub_handle_encoder)
 
     now = datetime.datetime.now()