rongo1 commited on
Commit
d6f9d21
·
1 Parent(s): 87132f1

chore: replaced stuff

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -197,7 +197,8 @@ Return format: [card1_data, card2_data, card3_data, ...]
197
  # Add metadata to each card's data
198
  logger.debug("Adding metadata to extracted data")
199
  for i, data in enumerate(response_data):
200
- data['method'] = model_name
 
201
  if i < len(filenames):
202
  data['filename'] = filenames[i]
203
  logger.debug(f"Added metadata to card {i+1}: {filenames[i]}")
@@ -230,7 +231,8 @@ Return format: [card1_data, card2_data, card3_data, ...]
230
  # Add metadata to each card's data
231
  logger.debug("Adding metadata to cleaned extracted data")
232
  for i, data in enumerate(response_data):
233
- data['method'] = model_name
 
234
  if i < len(filenames):
235
  data['filename'] = filenames[i]
236
  logger.debug(f"Added metadata to cleaned card {i+1}: {filenames[i]}")
 
197
  # Add metadata to each card's data
198
  logger.debug("Adding metadata to extracted data")
199
  for i, data in enumerate(response_data):
200
+ # Use user-friendly model name for Excel
201
+ data['method'] = "Speed-Optimized AI" if "flash" in model_name else "Accuracy-Optimized AI"
202
  if i < len(filenames):
203
  data['filename'] = filenames[i]
204
  logger.debug(f"Added metadata to card {i+1}: {filenames[i]}")
 
231
  # Add metadata to each card's data
232
  logger.debug("Adding metadata to cleaned extracted data")
233
  for i, data in enumerate(response_data):
234
+ # Use user-friendly model name for Excel
235
+ data['method'] = "Speed-Optimized AI" if "flash" in model_name else "Accuracy-Optimized AI"
236
  if i < len(filenames):
237
  data['filename'] = filenames[i]
238
  logger.debug(f"Added metadata to cleaned card {i+1}: {filenames[i]}")