Update app.py

app.py

@@ -14,12 +14,13 @@ import gradio as gr
 from rag_pipeline import RAGPipeline
 from adversarial_framework import *
 # Load all models and retrievers ONCE
-rag = RAGRetriever(
-    embedding_model="paraphrase-MiniLM-L3-v2",
-    cross_encoder_model="cross-encoder/ms-marco-MiniLM-L-6-v2",
-    generator_model="google/flan-t5-small"
+rag = RAGPipeline(
+    embedder_model: str = "infly/inf-retriever-v1-1.5b",
+    reranker_model: str = "cross-encoder/ms-marco-MiniLM-L-6-v2",
+    generator_model: str = "google/flan-t5-base"
 )
 
+
 adv_pipeline = AdversarialAttackPipeline(answer_generator=rag.generate_answer)
 
 # Define the Gradio wrapper