Spaces:

Shivani-Bhat
/

Topic-Modelling

Sleeping

Shivani-Bhat commited on Apr 26

Commit

f8b1ac8

verified ·

1 Parent(s): 89392e5

Update tools_additions.py

Files changed (1) hide show

tools_additions.py CHANGED Viewed

@@ -572,7 +572,7 @@ def _call_mistral(
 def _call_gemini(
     prompt: str,
     api_key: str,
-    model: str = "gemini-1.5-flash",
 ) -> str:
     """Call Google Gemini generateContent endpoint."""
     try:

 def _call_gemini(
     prompt: str,
     api_key: str,
+    model: str = "gemini-2.0-flash",
 ) -> str:
     """Call Google Gemini generateContent endpoint."""
     try: