3ie-intervention-outcome-entity-linking

Sleeping

CarlosMalaga commited on May 6, 2024

Commit

f0bf7d6

verified ·

1 Parent(s): 63a4e00

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -160,37 +160,37 @@ def load_model():
     retriever_question = GoldenRetriever(
         question_encoder="/home/user/app/models/retriever/question_encoder",
-        document_index="home/user/app/models/retriever/document_index/questions"
     )
     retriever_intervention = GoldenRetriever(
         question_encoder="/home/user/app/models/retriever/question_encoder",
-        document_index="home/user/app/models/retriever/document_index/interventions"
     )
     retriever_outcome = GoldenRetriever(
         question_encoder="/home/user/app/models/retriever/question_encoder",
-        document_index="home/user/app/models/retriever/document_index/outcomes"
     )
     retriever_question_db = GoldenRetriever(
         question_encoder="/home/user/app/models/retriever/question_encoder",
-        document_index="home/user/app/models/retriever/document_index/question_db"
     )
     retriever_intervention_db = GoldenRetriever(
         question_encoder="/home/user/app/models/retriever/question_encoder",
-        document_index="home/user/app/models/retriever/document_index/interventions_db"
     )
     retriever_outcome_db = GoldenRetriever(
         question_encoder="/home/user/app/models/retriever/question_encoder",
-        document_index="home/user/app/models/retriever/document_index/outcomes_db"
     )

     retriever_question = GoldenRetriever(
         question_encoder="/home/user/app/models/retriever/question_encoder",
+        document_index="/home/user/app/models/retriever/document_index/questions"
     )
     retriever_intervention = GoldenRetriever(
         question_encoder="/home/user/app/models/retriever/question_encoder",
+        document_index="/home/user/app/models/retriever/document_index/interventions"
     )
     retriever_outcome = GoldenRetriever(
         question_encoder="/home/user/app/models/retriever/question_encoder",
+        document_index="/home/user/app/models/retriever/document_index/outcomes"
     )
     retriever_question_db = GoldenRetriever(
         question_encoder="/home/user/app/models/retriever/question_encoder",
+        document_index="/home/user/app/models/retriever/document_index/question_db"
     )
     retriever_intervention_db = GoldenRetriever(
         question_encoder="/home/user/app/models/retriever/question_encoder",
+        document_index="/home/user/app/models/retriever/document_index/interventions_db"
     )
     retriever_outcome_db = GoldenRetriever(
         question_encoder="/home/user/app/models/retriever/question_encoder",
+        document_index="/home/user/app/models/retriever/document_index/outcomes_db"
     )