3ie-intervention-outcome-entity-linking

Sleeping

CarlosMalaga commited on May 16, 2024

Commit

846aa97

verified ·

1 Parent(s): bbfcd2c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -190,30 +190,30 @@ def load_model():
     )
     retriever_intervention_gpt_taxonomy = GoldenRetriever(
-        question_encoder="models/retriever/intervention/gpt/taxonomy/question_encoder",
-        document_index="models/retriever/intervention/gpt/taxonomy/document_index"
     )
     retriever_intervention_gpt_db = GoldenRetriever(
-        question_encoder="models/retriever/intervention/gpt/db/question_encoder",
-        document_index="models/retriever/intervention/gpt/db/document_index"
     )
     retriever_outcome_gpt_taxonomy = GoldenRetriever(
-        question_encoder="models/retriever/outcome/gpt/taxonomy/question_encoder",
-        document_index="models/retriever/outcome/gpt/taxonomy/document_index"
     )
     retriever_outcome_gpt_db = GoldenRetriever(
-        question_encoder="models/retriever/outcome/gpt/db/question_encoder",
-        document_index="models/retriever/outcome/gpt/db/document_index"
     )

     )
     retriever_intervention_gpt_taxonomy = GoldenRetriever(
+        question_encoder="models/retriever/intervention/gpt+llama/taxonomy/question_encoder",
+        document_index="models/retriever/intervention/gpt+llama/taxonomy/document_index"
     )
     retriever_intervention_gpt_db = GoldenRetriever(
+        question_encoder="models/retriever/intervention/gpt+llama/db/question_encoder",
+        document_index="models/retriever/intervention/gpt+llama/db/document_index"
     )
     retriever_outcome_gpt_taxonomy = GoldenRetriever(
+        question_encoder="models/retriever/outcome/gpt+llama/taxonomy/question_encoder",
+        document_index="models/retriever/outcome/gpt+llama/taxonomy/document_index"
     )
     retriever_outcome_gpt_db = GoldenRetriever(
+        question_encoder="models/retriever/outcome/gpt+llama/db/question_encoder",
+        document_index="models/retriever/outcome/gpt+llama/db/document_index"
     )