Update calvin.py

added preset for _engine in initialize method updated hyperparameter settings for complete_prompt method
pmohun · Jul 23, 2020 · c711f68 · c711f68
1 parent 3592e42
commit c711f68
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/calvin/calvin.py b/calvin/calvin.py
@@ -1,9 +1,9 @@
 import json
 import requests
 
-# #                                             #   #
-#   Calvin - abstraction module over openai's API  #
-# #                                             #   #
+# #                                               # #
+#   Calvin - abstraction module over openai's API   #
+# #                                               # #
 
 
 # Defile module-scoped variables
@@ -13,7 +13,7 @@
 
 
 # Initialize module
-def initialize(_api_key, _engine):
+def initialize(_api_key, _engine=engine):
   global engine, headers, prompts
 
   if _engine:
@@ -86,8 +86,8 @@ def create_completion(payload):
 def complete_prompt(prompt):
   payload = {
     "prompt": prompt,
-    "max_tokens": 5,
-    "temperature": 1,
+    "max_tokens": 150,
+    "temperature": 0.9,
     "top_p": 1,
     "n": 1
   }
@@ -143,4 +143,4 @@ def test_create_completion():
   "logprobs": null,
   "stop": "\n"
 }
-"""
+"""