Upload folder using huggingface_hub

app/app.py

@@ -49,5 +49,7 @@ if st.button("Run Data Cleaning API"):
             file_name='cleaned_df.csv',
             mime='text/csv',
         )
-            
-    
+    
+    # if both text_input and csv_file are empty, display error message
+    else:
+        st.write("Please enter text or upload a CSV file.")

app/openai_chat_completion.py

@@ -42,7 +42,14 @@ class OpenAIChatCompletions:
 
 
     # function to use test data to predict completions
-    def predict_jsonl(self, path_or_buf='../data/cookies_test.jsonl', n_samples=None, n_shot=None):
+    def predict_jsonl(
+        self,
+        # path_or_buf='../data/cookies_train.jsonl',
+        path_or_buf='~/data/cookies_train.jsonl',
+        n_samples=None,
+        n_shot=None
+        ):
+        
         jsonObj = pd.read_json(path_or_buf=path_or_buf, lines=True)
         if n_samples is not None:
             jsonObj = jsonObj.sample(n_samples, random_state=42)
@@ -76,5 +83,6 @@ class OpenAIChatCompletions:
         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)