Spaces:

kaleidoscope-data
/

data-cleaning-llm

Runtime error

cmagganas commited on Jul 11, 2023

Commit

d3c59b8

•

1 Parent(s): 07fd8f8

Upload folder using huggingface_hub

Files changed (1) hide show

app/dataclean_hf.py CHANGED Viewed

@@ -6,7 +6,7 @@ import numpy as np
 import streamlit as st
 from sqlalchemy import create_engine
-from yallmf.utils import run_with_timeout
 import openai
 from tqdm import tqdm
@@ -62,7 +62,7 @@ def clean_data(
     messages = [{'role':'system','content':'You are a helpful assistant. Return a properly-formatted CSV with the correct number of columns.'},
                 {'role':'user', 'content':p1+p2+'\n\nOutput CSV with header row:\n\n'}
                 ]
-    comp = run_with_timeout(openai.ChatCompletion.create,
         model='gpt-4',
         messages=messages,
         max_tokens=2000,

 import streamlit as st
 from sqlalchemy import create_engine
+# from yallmf.utils import run_with_timeout
 import openai
 from tqdm import tqdm
     messages = [{'role':'system','content':'You are a helpful assistant. Return a properly-formatted CSV with the correct number of columns.'},
                 {'role':'user', 'content':p1+p2+'\n\nOutput CSV with header row:\n\n'}
                 ]
+    comp = openai.ChatCompletion.create(
         model='gpt-4',
         messages=messages,
         max_tokens=2000,