updated db config

Author: WaseeA

README.md

@@ -14,7 +14,7 @@ docker run \
   -d \
   --name trino-nlp-embeddings \
   -p 8080:8080 \
-  waseedockerhub9/trino-nlp-embeddings:0.53
+  waseedockerhub9/trino-nlp-embeddings:0.67
 ```
 
 Then use your favourite SQL client to connect to Trino running at http://localhost:8080

src/py_scripts/requirements.txt

@@ -1,7 +1,3 @@
-# numpy==1.26.4
-# torch==2.0.1+cpu
-# -f https://download.pytorch.org/whl/cpu/torch_stable.html
-# git+https://github.com/huggingface/transformers
 sentence-transformers==3.3.1
 psycopg2-binary==2.9.10
-scipy==1.13.1
+scipy==1.14.1

src/py_scripts/script.py

@@ -13,16 +13,14 @@
 #
 
 import sys
-# import torch
 import psycopg2
-# import numpy as np
 from transformers import BertTokenizer, BertModel
 from sentence_transformers import SentenceTransformer
 from scipy.spatial.distance import cosine
 
 # Define database connection parameters
 DB_CONFIG = {
-    "host": "localhost",
+    "host": "some-postgres", # same name as your docker container running your postgres instance
     "port": "5432",
     "database": "postgres",
     "user": "postgres",