lhoestq HF staff commited on
Commit
1fa40b1
·
1 Parent(s): 4d5787c

fix decimal values

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -202,7 +202,7 @@ with gr.Blocks(css=css) as demo:
202
  prompt=prompt,
203
  )}]
204
  response_format = {"type": "json", "value": {"properties": {"data": {"type": "array", "items": format, "minItems": len(rows), "maxItems": len(rows)}}, "required": ["data"]}}
205
- yield from ijson.items(StringIteratorIO(stream_reponse(messages, response_format=response_format)), "data.item", buf_size=4)
206
 
207
 
208
  def stream_rewrite_dataset_row_by_row(dataset: str, rows: list[dict[str, str]], prompt: str, format: str, input_preview_rows: list[dict[str, str]], output_preview_rows: list[dict[str, str]]) -> Iterator[dict[str, str]]:
@@ -216,7 +216,7 @@ with gr.Blocks(css=css) as demo:
216
  )}]
217
  response_format = {"type": "json", "value": {"properties": {"data": {"type": "array", "items": format, "minItems": len(rows), "maxItems": len(rows)}}, "required": ["data"]}}
218
  try:
219
- yield from ijson.items(StringIteratorIO(stream_reponse(messages, response_format=response_format)), "data.item", buf_size=4)
220
  except ijson.IncompleteJSONError as e:
221
  print(f"{type(e).__name__}: {e}")
222
  print("Warning: Some rows were missing during ReWriting.")
 
202
  prompt=prompt,
203
  )}]
204
  response_format = {"type": "json", "value": {"properties": {"data": {"type": "array", "items": format, "minItems": len(rows), "maxItems": len(rows)}}, "required": ["data"]}}
205
+ yield from ijson.items(StringIteratorIO(stream_reponse(messages, response_format=response_format)), "data.item", buf_size=4, use_float=True)
206
 
207
 
208
  def stream_rewrite_dataset_row_by_row(dataset: str, rows: list[dict[str, str]], prompt: str, format: str, input_preview_rows: list[dict[str, str]], output_preview_rows: list[dict[str, str]]) -> Iterator[dict[str, str]]:
 
216
  )}]
217
  response_format = {"type": "json", "value": {"properties": {"data": {"type": "array", "items": format, "minItems": len(rows), "maxItems": len(rows)}}, "required": ["data"]}}
218
  try:
219
+ yield from ijson.items(StringIteratorIO(stream_reponse(messages, response_format=response_format)), "data.item", buf_size=4, use_float=True)
220
  except ijson.IncompleteJSONError as e:
221
  print(f"{type(e).__name__}: {e}")
222
  print("Warning: Some rows were missing during ReWriting.")