Kazel commited on
Commit
6458211
·
1 Parent(s): e18302f
Files changed (2) hide show
  1. rag.py +1 -1
  2. requirements.txt +1 -1
rag.py CHANGED
@@ -75,7 +75,7 @@ class Rag:
75
  images = [Image.open(path) for path in imagePaths]
76
 
77
  response = client.models.generate_content(
78
- model="gemini-2.5-flash",
79
  contents=[images, query],
80
  )
81
 
 
75
  images = [Image.open(path) for path in imagePaths]
76
 
77
  response = client.models.generate_content(
78
+ model="gemini-2.5-pro",
79
  contents=[images, query],
80
  )
81
 
requirements.txt CHANGED
@@ -22,7 +22,7 @@ pdf2image
22
  spaces>=0.0.1
23
  ollama
24
  # Google Gemini API
25
- google-generativeai>=0.3.0
26
 
27
  # Additional utilities
28
  psutil>=5.9.0
 
22
  spaces>=0.0.1
23
  ollama
24
  # Google Gemini API
25
+ google-genai
26
 
27
  # Additional utilities
28
  psutil>=5.9.0