upd: fix keys

microsoft · Dec 22, 2023 · b6210b3 · b6210b3
1 parent cf3988a
commit b6210b3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/promptbench/models/__init__.py b/promptbench/models/__init__.py
@@ -159,5 +159,5 @@ def __call__(self, input_text, **kwargs):
 
 
 if __name__ == "__main__":
-    model = LLMModel(model='gemini-pro', gemini_key="AIzaSyBvb3QX6u2ToRTm9NpWnZrXOkJHhj6gr0o")
+    model = LLMModel(model='gemini-pro', gemini_key="xx")
     print(model("Hello, my name is"))