updated data extraction and readme

feast-dev · May 2, 2024 · 064dce0 · 064dce0
1 parent 7055939
commit 064dce0
Show file tree

Hide file tree

Showing 5 changed files with 109 additions and 18 deletions.
diff --git a/module_4_rag/README.md b/module_4_rag/README.md
@@ -21,6 +21,16 @@ Then run
 ```bash
 python batch_score_documents.py
 ```
+Which will output data to `city_wikipedia_summaries_with_embeddings.parquet`
+
+## Feast
+
+To get started, make sure to have Feast installed and PostGreSQL.
+
+First run
+```bash
+feast apply
+```
 
 # Overview
 

diff --git a/module_4_rag/batch_score_documents.py b/module_4_rag/batch_score_documents.py
@@ -1,4 +1,4 @@
-import os 
+import os
 import pandas as pd
 from transformers import AutoTokenizer, AutoModel
 import torch