ShiwenNi commited on
Commit
bd30d4d
·
verified ·
1 Parent(s): 3b21f9e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -85,7 +85,7 @@ class Reviewer:
85
  ]
86
  try:
87
  response = openai.ChatCompletion.create(
88
- model="gpt-3.5-turbo-16k",
89
  messages=messages,
90
  temperature=0.5
91
  )
@@ -109,8 +109,6 @@ class Reviewer:
109
 
110
 
111
  def extract_chapter(self, pdf_path):
112
- if isinstance(pdf_path, NamedString):
113
- pdf_path = BytesIO(pdf_path.name) # 将NamedString对象转换为BytesIO对象
114
  file_object = BytesIO(pdf_path)
115
  pdf_reader = PyPDF2.PdfReader(file_object)
116
  # 获取PDF的总页数
 
85
  ]
86
  try:
87
  response = openai.ChatCompletion.create(
88
+ model="gpt-4o-mini",
89
  messages=messages,
90
  temperature=0.5
91
  )
 
109
 
110
 
111
  def extract_chapter(self, pdf_path):
 
 
112
  file_object = BytesIO(pdf_path)
113
  pdf_reader = PyPDF2.PdfReader(file_object)
114
  # 获取PDF的总页数