Spaces:

chansung
/

paper_qa

Running on CPU Upgrade

chansung commited on Mar 15

Commit

6e0c0c0

•

1 Parent(s): f46ecbb

Update gen/gemini.py

Files changed (1) hide show

gen/gemini.py CHANGED Viewed

@@ -10,9 +10,9 @@ from gen.utils import parse_first_json_snippet
 def determine_model_name(given_image=None):
   if given_image is None:
-    return "gemini-pro"
   else:
-    return "gemini-pro-vision"
 def construct_image_part(given_image):
   return {

 def determine_model_name(given_image=None):
   if given_image is None:
+    return "gemini-1.0-pro"
   else:
+    return "gemini-1.0-pro-vision"
 def construct_image_part(given_image):
   return {