Spaces:
Build error
Build error
Commit
·
fad460a
1
Parent(s):
ae3526d
Update app.py
Browse files
app.py
CHANGED
|
@@ -30,12 +30,12 @@ def generate_filename(prompt, file_type):
|
|
| 30 |
return f"{safe_date_time}_{safe_prompt}.{file_type}"
|
| 31 |
|
| 32 |
def chat_with_model(prompt, document_section):
|
| 33 |
-
|
| 34 |
conversation = [{'role': 'system', 'content': 'You are a helpful assistant.'}]
|
| 35 |
conversation.append({'role': 'user', 'content': prompt})
|
| 36 |
if len(document_section)>0:
|
| 37 |
conversation.append({'role': 'assistant', 'content': document_section})
|
| 38 |
-
response = openai.ChatCompletion.create(model=
|
| 39 |
return response
|
| 40 |
#return response['choices'][0]['message']['content']
|
| 41 |
|
|
|
|
| 30 |
return f"{safe_date_time}_{safe_prompt}.{file_type}"
|
| 31 |
|
| 32 |
def chat_with_model(prompt, document_section):
|
| 33 |
+
model = model_choice
|
| 34 |
conversation = [{'role': 'system', 'content': 'You are a helpful assistant.'}]
|
| 35 |
conversation.append({'role': 'user', 'content': prompt})
|
| 36 |
if len(document_section)>0:
|
| 37 |
conversation.append({'role': 'assistant', 'content': document_section})
|
| 38 |
+
response = openai.ChatCompletion.create(model=model, messages=conversation)
|
| 39 |
return response
|
| 40 |
#return response['choices'][0]['message']['content']
|
| 41 |
|