Spaces:
Running
Running
Update query_classifier.py
Browse files- 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-
|
| 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',
|