Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -15,9 +15,9 @@ from rag_pipeline import RAGPipeline
|
|
15 |
from adversarial_framework import *
|
16 |
# Load all models and retrievers ONCE
|
17 |
rag = RAGPipeline(
|
18 |
-
embedder_model
|
19 |
-
reranker_model
|
20 |
-
generator_model
|
21 |
)
|
22 |
|
23 |
|
|
|
15 |
from adversarial_framework import *
|
16 |
# Load all models and retrievers ONCE
|
17 |
rag = RAGPipeline(
|
18 |
+
embedder_model = "infly/inf-retriever-v1-1.5b",
|
19 |
+
reranker_model = "cross-encoder/ms-marco-MiniLM-L-6-v2",
|
20 |
+
generator_model = "google/flan-t5-base"
|
21 |
)
|
22 |
|
23 |
|