feat: add tranning in start container

ingest.py

@@ -19,7 +19,6 @@ from constants import (
     SOURCE_DIRECTORY,
 )
 
-
 def load_single_document(file_path: str) -> Document:
     # Loads a single document from a file path
     file_extension = os.path.splitext(file_path)[1]
@@ -134,8 +133,9 @@ def main(device_type):
     # Create embeddings
     embeddings = HuggingFaceInstructEmbeddings(
         model_name=EMBEDDING_MODEL_NAME,
-        model_kwargs={"device": device_type},
+        model_kwargs={"device": "cuda"},
     )
+
     # change the embedding type here if you are running into issues.
     # These are much smaller embeddings and will work for most appications
     # If you use HuggingFaceEmbeddings, make sure to also use the same in the
@@ -159,6 +159,3 @@ if __name__ == "__main__":
         format="%(asctime)s - %(levelname)s - %(filename)s:%(lineno)s - %(message)s", level=logging.INFO
     )
     main()
-    # run_localGPT_API.main()
-    # localGPTUI.main()
-

run.sh

@@ -1,4 +1,7 @@
-# Redis Support uncomment this lines
+#!/bin/sh
+
 nohup redis-server &
 
+python ingest.py
+
 uvicorn "main:app" --port 7860 --host 0.0.0.0