Sage
commited on
Commit
•
8b0e4a2
1
Parent(s):
6fc7f47
settings changes
Browse files
app.py
CHANGED
@@ -32,6 +32,7 @@ def chat_gpt_image(content, context):
|
|
32 |
completion = openai.ChatCompletion.create(
|
33 |
model=gpt_model,
|
34 |
user="1",
|
|
|
35 |
messages=[
|
36 |
{"role": "system", "content": prompt},
|
37 |
{"role": "user", "content": final_content}
|
@@ -74,6 +75,7 @@ def chat_gpt_document(content, document_type, context):
|
|
74 |
completion_1 = openai.ChatCompletion.create(
|
75 |
model=gpt_model,
|
76 |
user="1",
|
|
|
77 |
messages=[
|
78 |
{"role": "system", "content": prompt},
|
79 |
{"role": "user", "content": content_1}
|
@@ -193,24 +195,24 @@ def batch_document(content, document_type, context, progress=gr.Progress()):
|
|
193 |
data = json.loads(chat_gpt_document(detect_document(x),document_type,context))
|
194 |
combined_data.append(data)
|
195 |
break
|
196 |
-
except
|
197 |
-
logging.error(f'Retry {i+1} failed:
|
198 |
if i < retries - 1:
|
199 |
logging.error(f'Retrying in {timeout} seconds...')
|
200 |
time.sleep(timeout)
|
201 |
i += 1
|
202 |
-
except openai.error.RateLimitError:
|
203 |
-
|
204 |
-
|
205 |
-
|
206 |
-
|
207 |
-
|
208 |
logging.info(combined_data)
|
209 |
return save_json(combined_data, document_type)
|
210 |
|
211 |
def save_json(text, filename):
|
212 |
filename = filename+".json"
|
213 |
-
with open(filename, "w") as outfile:
|
214 |
json.dump(text, outfile)
|
215 |
return filename
|
216 |
|
|
|
32 |
completion = openai.ChatCompletion.create(
|
33 |
model=gpt_model,
|
34 |
user="1",
|
35 |
+
temperature="0.5",
|
36 |
messages=[
|
37 |
{"role": "system", "content": prompt},
|
38 |
{"role": "user", "content": final_content}
|
|
|
75 |
completion_1 = openai.ChatCompletion.create(
|
76 |
model=gpt_model,
|
77 |
user="1",
|
78 |
+
temperature="0.1",
|
79 |
messages=[
|
80 |
{"role": "system", "content": prompt},
|
81 |
{"role": "user", "content": content_1}
|
|
|
195 |
data = json.loads(chat_gpt_document(detect_document(x),document_type,context))
|
196 |
combined_data.append(data)
|
197 |
break
|
198 |
+
except Exception as e:
|
199 |
+
logging.error(f'Retry {i+1} failed: ', e)
|
200 |
if i < retries - 1:
|
201 |
logging.error(f'Retrying in {timeout} seconds...')
|
202 |
time.sleep(timeout)
|
203 |
i += 1
|
204 |
+
# except openai.error.RateLimitError:
|
205 |
+
# logging.error(f'Retry {j+1} failed: openai.error.RateLimitError')
|
206 |
+
# if j < retries - 1:
|
207 |
+
# logging.error(f'Retrying in {timeout} seconds...')
|
208 |
+
# time.sleep(timeout)
|
209 |
+
# j += 1
|
210 |
logging.info(combined_data)
|
211 |
return save_json(combined_data, document_type)
|
212 |
|
213 |
def save_json(text, filename):
|
214 |
filename = filename+".json"
|
215 |
+
with open(filename, "w", encoding='utf-8') as outfile:
|
216 |
json.dump(text, outfile)
|
217 |
return filename
|
218 |
|