[update] gemini model inference

microsoft · Jan 1, 2024 · 386cd66 · 386cd66
1 parent 935ad6a
commit 386cd66
Showing 1 changed file with 4 additions and 5 deletions.
diff --git a/promptbench/models/models.py b/promptbench/models/models.py
@@ -523,8 +523,7 @@ def predict(self, input_text, **kwargs):
                                     generation_config=generation_config,
                                     safety_settings=safety_settings)
 
-        try:
-            response = model.generate_content(input_text).text
-        except:
-            response = "error!"
-        return response
+
+        response = model.generate_content(input_text).text
+
+        return response