chibuzordev commited on
Commit
679bed8
·
verified ·
1 Parent(s): ad4fb63

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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: str = "infly/inf-retriever-v1-1.5b",
19
- reranker_model: str = "cross-encoder/ms-marco-MiniLM-L-6-v2",
20
- generator_model: str = "google/flan-t5-base"
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