Update app.py

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 from smolagents.agents import ToolCallingAgent
-from smolagents import tool, GeminiModel  # Assuming GeminiModel is the correct class
+from smolagents import tool, LiteLLMModel
 from typing import Optional
 import cv2
 import pytesseract
@@ -8,8 +8,8 @@ from PIL import Image
 import io
 import numpy as np
 
-# Define the GeminiModel
-model = GeminiModel(api_key="AIzaSyCuK0Z50g5AGZrpvG54c8v_8rsOSRPjJJs", model_id="gemini-1.5-flash")
+# Define the LiteLLMModel with OpenAI key
+model = LiteLLMModel(model_id="gpt-4o", api_key="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")
 
 @tool
 def extract_components(image_data: bytes) -> str: