Update app.py

app.py

@@ -45,13 +45,18 @@ def create_custom_index():
     dataset_path = "icici_bank_dataset"
     index_path = "icici_bank_index"
     dataset.save_to_disk(dataset_path)
+    print(f"Dataset saved to {dataset_path}")
+    
+    # Add FAISS index
     dataset.add_faiss_index("text")
     dataset.get_index("text").save(index_path)
+    print(f"FAISS index saved to {index_path}")
     
     return dataset_path, index_path
 
 # Load the fine-tuned RAG model and tokenizer
 tokenizer = RagTokenizer.from_pretrained("facebook/rag-sequence-base")
+print("Tokenizer loaded successfully.")
 
 try:
     # Create and save the custom index
@@ -64,9 +69,11 @@ try:
         passages_path=dataset_path,
         index_path=index_path
     )
+    print("Retriever loaded successfully.")
     
     # Load the RAG model
     model = RagSequenceForGeneration.from_pretrained("facebook/rag-sequence-base", retriever=retriever)
+    print("Model loaded successfully.")
 except Exception as e:
     print(f"Error initializing model or retriever: {e}")
     retriever = None