Update models_tokens.py

MobinX · web-flow · commit 7f30da993089 · 2024-05-18T23:32:27.000+06:00
diff --git a/scrapegraphai/helpers/models_tokens.py b/scrapegraphai/helpers/models_tokens.py
@@ -27,6 +27,8 @@
     },
     "gemini": {
         "gemini-pro": 128000,
+        "gemini-1.5-flash-latest":128000,
+        "gemini-1.5-pro-latest":128000,
         "models/embedding-001": 2048
     },
 
