Spaces:

AlexSychovUN
/

BindingAffinityPrediction

Sleeping

AlexSychovUN commited on Jan 27

Commit

a21e922

1 Parent(s): 891f712

Prepared for deploy

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -236,7 +236,7 @@ def get_gemini_explanation(
     for attempt in range(max_retries):
         try:
             response = client.models.generate_content(
-                model="gemini-1.5-flash", contents=prompt
             )
             return response.text

     for attempt in range(max_retries):
         try:
             response = client.models.generate_content(
+                model="gemini-2.5-flash-lite", contents=prompt
             )
             return response.text