jdmorzan commited on
Commit
45bc3a6
·
verified ·
1 Parent(s): c434c71

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -24,8 +24,9 @@ def classify_text_chat(template, work, exam):
24
  ],
25
  max_tokens=1000,
26
  temperature=0)
27
- #return response['choices'][0]['message']['content'].strip()
28
- return response
 
29
 
30
  def read_docx(file_path):
31
  loader = Docx2txtLoader(file_path)
 
24
  ],
25
  max_tokens=1000,
26
  temperature=0)
27
+ choices = response.choices[0]
28
+ text = choices.message.content
29
+ return text
30
 
31
  def read_docx(file_path):
32
  loader = Docx2txtLoader(file_path)