Evil-paradox007 commited on
Commit
97287e8
·
verified ·
1 Parent(s): 1b410bd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ import google.generativeai as genai
29
  DATA_DIR = os.environ.get("DATA_DIR", "/data")
30
  GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY", "")
31
  EMBED_MODEL_NAME = "all-MiniLM-L6-v2"
32
- GEMINI_MODEL = "gemini-1.5-flash"
33
  TOP_K = 5 # Number of chunks to retrieve
34
  GEMINI_TIMEOUT = 30 # Seconds before Gemini API request times out
35
 
 
29
  DATA_DIR = os.environ.get("DATA_DIR", "/data")
30
  GEMINI_API_KEY = os.environ.get("GEMINI_API_KEY", "")
31
  EMBED_MODEL_NAME = "all-MiniLM-L6-v2"
32
+ GEMINI_MODEL = "gemini-2.0-flash"
33
  TOP_K = 5 # Number of chunks to retrieve
34
  GEMINI_TIMEOUT = 30 # Seconds before Gemini API request times out
35