eaglelandsonce commited on
Commit
fad460a
·
1 Parent(s): ae3526d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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=model_choice)
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