mlbench123 commited on
Commit
d5384be
·
verified ·
1 Parent(s): 893b07b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -152,7 +152,7 @@ def process_files_for_gemini(files: List[str]) -> Dict[str, Any]:
152
  return processed_data
153
 
154
 
155
- def extract_with_gemini(processed_data: Dict[str, Any], api_key: str, model_name: str = "gemini-1.5-flash") -> Dict[str, Any]:
156
  """Extract structured data using Gemini with enhanced multimodal processing"""
157
 
158
  if not api_key or api_key.strip() == "":
@@ -235,7 +235,7 @@ def process_documents(files):
235
 
236
  # Use the hardcoded API key and default model
237
  api_key = GEMINI_API_KEY
238
- model_choice = "gemini-1.5-flash"
239
 
240
  if not api_key or api_key.strip() == "":
241
  return "❌ Error: API key not configured in code", "{}", "API key missing"
 
152
  return processed_data
153
 
154
 
155
+ def extract_with_gemini(processed_data: Dict[str, Any], api_key: str, model_name: str = "gemini-2.0-flash") -> Dict[str, Any]:
156
  """Extract structured data using Gemini with enhanced multimodal processing"""
157
 
158
  if not api_key or api_key.strip() == "":
 
235
 
236
  # Use the hardcoded API key and default model
237
  api_key = GEMINI_API_KEY
238
+ model_choice = "gemini-2.0-flash"
239
 
240
  if not api_key or api_key.strip() == "":
241
  return "❌ Error: API key not configured in code", "{}", "API key missing"