Update app.py

app.py

@@ -41,7 +41,11 @@ def extract_text_from_docx(file):
 def generate_financial_statements(text):
     # Load RAG model and tokenizer
     tokenizer = RagTokenizer.from_pretrained("facebook/rag-sequence-base")
-    retriever = RagRetriever.from_pretrained("facebook/rag-sequence-base", index_name="exact")
+    retriever = RagRetriever.from_pretrained(
+        "facebook/rag-sequence-base",
+        index_name="exact",
+        trust_remote_code=True  # Allow execution of remote code
+    )
     model = RagSequenceForGeneration.from_pretrained("facebook/rag-sequence-base", retriever=retriever)
 
     # Tokenize input text
@@ -129,6 +133,7 @@ st.sidebar.write("""
 - python-docx
 - pandas
 - faiss-cpu (or faiss-gpu)
+- datasets
 """)
 
 # Deployment instructions