WIP - Streamlit UI, firecrawl - V0.5

2024-06-13 17:43:57 +05:30
parent 8dfcb1f536
commit 128b6f3878
6 changed files with 196 additions and 40 deletions
--- a/lib/gpt_providers/text_generation/gemini_pro_text.py
+++ b/lib/gpt_providers/text_generation/gemini_pro_text.py
@@ -29,7 +29,7 @@ def gemini_text_response(prompt, temperature, top_p, n, max_tokens):
    except Exception as err:
        logger.error(f"Failed to configure Gemini: {err}")
    logger.info(f"Temp: {temperature}, MaxTokens: {max_tokens}, TopP: {top_p}, N: {n}")
-    # Set up the model
+    # Set up AI model config
    generation_config = {
        "temperature": temperature,
        "top_p": top_p,