Update app.py

app.py

@@ -3,6 +3,14 @@ import json
 import firebase_admin
 from firebase_admin import credentials, db
 from transformers import RagTokenizer, RagRetriever, RagSequenceForGeneration
+from transformers import RagRetriever
+
+retriever = RagRetriever.from_pretrained(
+    "facebook/rag-token-base", 
+    use_dummy_dataset=True, 
+    trust_remote_code=True
+)
+
 import gradio as gr
 
 # Initialize Firebase Admin SDK