Spaces:

Wall06
/

lead-qualifier-api

Sleeping

Wall06 commited on 10 days ago

Commit

f931a89

verified ·

1 Parent(s): aaf704f

Upload folder using huggingface_hub

Files changed (1) hide show

ai_engine.py CHANGED Viewed

@@ -7,7 +7,7 @@ genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 # Initialize the model (using the standard gemini-pro which is universally supported)
 model = genai.GenerativeModel(
-    model_name='gemini-1.5-flash-latest',
     generation_config={"response_mime_type": "application/json"} # Forces perfect JSON!
 )

 # Initialize the model (using the standard gemini-pro which is universally supported)
 model = genai.GenerativeModel(
+    model_name='gemini-pro',
     generation_config={"response_mime_type": "application/json"} # Forces perfect JSON!
 )