Upload folder using huggingface_hub

app/openai_chat_completion.py

@@ -15,7 +15,7 @@ class OpenAIChatCompletions:
         self.model = model
         self.system_message = system_message
 
-  
+
     # function to input args such as model, prompt, etc. and return completion
     def openai_chat_completion(self, prompt, n_shot=None):
         messages = [{"role": "system", "content": self.system_message}] if self.system_message else []
@@ -71,6 +71,10 @@ class OpenAIChatCompletions:
 
     # a method that samples n rows from a jsonl file, returning a pandas dataframe
     @staticmethod
-    def _sample_jsonl(path_or_buf='../data/cookies_train.jsonl', n_samples=5):
+    def _sample_jsonl(
+        # path_or_buf='../data/cookies_train.jsonl',
+        path_or_buf='~/data/cookies_train.jsonl',
+        n_samples=5
+        ):
         jsonObj = pd.read_json(path_or_buf=path_or_buf, lines=True)
         return jsonObj.sample(n_samples, random_state=42)