prashantmatlani commited on
Commit
74d91ad
·
1 Parent(s): aae6d78

updated model for OCR

Browse files
Files changed (1) hide show
  1. perception_agent.py +3 -1
perception_agent.py CHANGED
@@ -8,6 +8,8 @@ from pypdf import PdfReader # Cleanly leverages your requirements.txt package
8
  from groq import Groq
9
  from agent_logging import log_agent_action
10
 
 
 
11
  client = Groq(api_key=os.getenv("GROQ_API_KEY"))
12
 
13
  def read_image_file(file_path):
@@ -22,7 +24,7 @@ def read_image_file(file_path):
22
  mime_type = f"image/{ext}" if ext != "jpg" else "image/jpeg"
23
 
24
  response = client.chat.completions.create(
25
- model="llama-3.2-11b-vision-preview",
26
  messages=[
27
  {
28
  "role": "user",
 
8
  from groq import Groq
9
  from agent_logging import log_agent_action
10
 
11
+ #model = "llama-3.2-11b-vision-preview" # -> Decommissioned
12
+ model = "llama-4-scout-17b-16e-instruct"
13
  client = Groq(api_key=os.getenv("GROQ_API_KEY"))
14
 
15
  def read_image_file(file_path):
 
24
  mime_type = f"image/{ext}" if ext != "jpg" else "image/jpeg"
25
 
26
  response = client.chat.completions.create(
27
+ model=model,
28
  messages=[
29
  {
30
  "role": "user",