Svngoku commited on
Commit
733d98b
·
verified ·
1 Parent(s): 6b0ebea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -140,7 +140,7 @@ class OCRProcessor:
140
  base64_url = f"data:image/jpeg;base64,{encoded_image}"
141
  ocr_response = self._call_ocr_api({"type": "image_url", "image_url": base64_url})
142
  markdown = self._extract_markdown(ocr_response)
143
-
144
  chat_response = self._call_chat_complete(
145
  model="pixtral-12b-latest",
146
  messages=[{
@@ -156,18 +156,17 @@ class OCRProcessor:
156
  response_format={"type": "json_object"},
157
  temperature=0
158
  )
159
-
160
  # Ensure the response is a dictionary
161
  response_content = chat_response.choices[0].message.content
162
  if isinstance(response_content, list):
163
  response_content = response_content[0] if response_content else "{}"
164
-
165
  content = json.loads(response_content)
166
  return self._format_structured_response(temp_path, content)
167
  except Exception as e:
168
  return self._handle_error("structured OCR", e)
169
 
170
-
171
  @staticmethod
172
  def _extract_markdown(response: OCRResponse) -> str:
173
  return response.pages[0].markdown if response.pages else "No text extracted"
 
140
  base64_url = f"data:image/jpeg;base64,{encoded_image}"
141
  ocr_response = self._call_ocr_api({"type": "image_url", "image_url": base64_url})
142
  markdown = self._extract_markdown(ocr_response)
143
+
144
  chat_response = self._call_chat_complete(
145
  model="pixtral-12b-latest",
146
  messages=[{
 
156
  response_format={"type": "json_object"},
157
  temperature=0
158
  )
159
+
160
  # Ensure the response is a dictionary
161
  response_content = chat_response.choices[0].message.content
162
  if isinstance(response_content, list):
163
  response_content = response_content[0] if response_content else "{}"
164
+
165
  content = json.loads(response_content)
166
  return self._format_structured_response(temp_path, content)
167
  except Exception as e:
168
  return self._handle_error("structured OCR", e)
169
 
 
170
  @staticmethod
171
  def _extract_markdown(response: OCRResponse) -> str:
172
  return response.pages[0].markdown if response.pages else "No text extracted"