upload

Dockerfile

@@ -6,8 +6,10 @@ WORKDIR /app
 
 # make directory /app/files/
 RUN mkdir /app/files/
+RUN mkdir /app/ckpt/
 # grant permission to write to /app/files/
 RUN chmod 777 /app/files/ 
+RUN chmod 777 /app/ckpt/
 
 # Copy the requirements file into the container
 COPY requirements.txt .

base_class.py

@@ -19,7 +19,7 @@ class Embedding_Model(metaclass=abc.ABCMeta):
 
     def __init__(self, model_name) -> None:
         """Initialize the embedding model"""
-        embedding_cache_path = f"/data/ckpt/embedding_cache_{model_name}.pkl"
+        embedding_cache_path = f"/app/ckpt/embedding_cache_{model_name}.pkl"
         self.embedding_cache_path = embedding_cache_path
 
         # load the cache if it exists, and save a copy to disk
@@ -43,7 +43,7 @@ class AbstractPDFParser(metaclass=abc.ABCMeta):
 
     def __init__(self, db_name) -> None:
         """Initialize the pdf database"""
-        db_cache_path = f"/data/ckpt/pdf_parser_{db_name}.pkl"
+        db_cache_path = f"/app/ckpt/pdf_parser_{db_name}.pkl"
         self.db_cache_path = db_cache_path
 
         # load the cache if it exists, and save a copy to disk