Update llm/huggingfacehub/hf_model.py

llm/huggingfacehub/hf_model.py

@@ -26,12 +26,16 @@ logger.addHandler(file_handler)
 
 logger.info("Getting information from hf_model module")
 
-try:
-   os.chdir('/home/user/app/llm/')
-except FileNotFoundError:
-   print("Error: Could not move up. You might be at the root directory.")
+# try:
+#    os.chdir('/home/user/app/llm/')
+# except FileNotFoundError:
+#    print("Error: Could not move up. You might be at the root directory.")
 
-work_dir = os.getcwd() 
+# work_dir = os.getcwd() 
+
+print("CWD : ", os.getcwd())
+
+llm_dir = '/home/user/app/llm/'
 
 
 class HF_Mistaril(HFInterface, ABC):
@@ -53,7 +57,7 @@ class HF_Mistaril(HFInterface, ABC):
     @staticmethod
     def __read_yaml():
         try:
-            yaml_file = os.path.join(work_dir, 'prompts.yaml')
+            yaml_file = os.path.join(llm_dir, 'prompts.yaml')
             with open(yaml_file, 'r') as f:
                 data = yaml.safe_load(f)
             f.close()
@@ -101,7 +105,7 @@ class HF_TinyLlama(HFInterface, ABC):
     @staticmethod
     def __read_yaml():
         try:
-            yaml_file = os.path.join(work_dir, 'prompts.yaml')
+            yaml_file = os.path.join(llm_dir, 'prompts.yaml')
             with open(yaml_file, 'r') as f:
                 data = yaml.safe_load(f)
             f.close()