From 386cd660602ed26731cef76213222509626f57ce Mon Sep 17 00:00:00 2001
From: Kaijie Zhu <31989262+Immortalise@users.noreply.github.com>
Date: Mon, 1 Jan 2024 22:04:36 +0800
Subject: [PATCH] [update] gemini model inference

---
 promptbench/models/models.py | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/promptbench/models/models.py b/promptbench/models/models.py
index f59ec01..7dcb282 100644
--- 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
\ No newline at end of file
+
+        response = model.generate_content(input_text).text
+
+        return response