Spaces:

AnshulS
/

endpointwebappshl

Sleeping

AnshulS commited on May 8

Commit

06b8d86

verified ·

1 Parent(s): 9e9d5ee

Update reranker.py

Files changed (1) hide show

reranker.py CHANGED Viewed

@@ -3,7 +3,7 @@ import google.generativeai as genai
 import json
 genai.configure(api_key=os.environ["GEMINI_API_KEY"])
-model = genai.GenerativeModel("models/gemini-1.5-flash")
 def rerank(query, candidates):
     prompt = f"""

 import json
 genai.configure(api_key=os.environ["GEMINI_API_KEY"])
+model = genai.GenerativeModel("models/gemini-2.0-flash")
 def rerank(query, candidates):
     prompt = f"""