Rivalcoder commited on
Commit
1a1eddd
·
verified ·
1 Parent(s): 2497a16

Update query_classifier.py

Browse files
Files changed (1) hide show
  1. query_classifier.py +1 -1
query_classifier.py CHANGED
@@ -9,7 +9,7 @@ genai.configure(api_key=os.getenv('GOOGLE_API_KEY'))
9
 
10
  class QueryClassifier:
11
  def __init__(self):
12
- self.model = genai.GenerativeModel('gemini-1.5-flash')
13
  self.process_mapping = {
14
  'legal_aid': 'Legal Aid Application',
15
  'consumer_complaint': 'Consumer Complaint Filing',
 
9
 
10
  class QueryClassifier:
11
  def __init__(self):
12
+ self.model = genai.GenerativeModel('gemini-2.5-flash-lite')
13
  self.process_mapping = {
14
  'legal_aid': 'Legal Aid Application',
15
  'consumer_complaint': 'Consumer Complaint Filing',