temperature added

SynonymEditor.py

@@ -7,10 +7,11 @@ nltk.download('punkt')
 
 class SynonymEditor:
 
-    def __init__(self, api_key, model_engine, max_tokens):
+    def __init__(self, api_key, model_engine, max_tokens, temperature):
         openai.api_key = api_key
         self.model_engine = model_engine
         self.max_tokens = max_tokens
+        self.temperature = temperature
 
     # Play with the prompts here and change the return index to change and see the effect of the prompt on the output quality
     # Note that the longer the prompt, higher the token used and hence the billing
@@ -27,7 +28,7 @@ class SynonymEditor:
         response = openai.Completion.create(
             model=self.model_engine,
             prompt=prompt,
-            temperature=0.6,
+            temperature=self.temperature,
             max_tokens=self.max_tokens,
             top_p=1,
             frequency_penalty=0,

app.py

@@ -2,10 +2,10 @@ import gradio as gr
 from SynonymEditor import SynonymEditor
 
 
-def replace_synonyms(api_key, text):
+def replace_synonyms(api_key, text, temperature):
     model_engine = "text-davinci-003"
     max_tokens = 500
-    editor = SynonymEditor(api_key, model_engine, max_tokens)
+    editor = SynonymEditor(api_key, model_engine, max_tokens, temperature)
     return editor._edit_text(text)
 
 
@@ -14,8 +14,11 @@ api_key = gr.inputs.Textbox(label="API Key", lines=1, default="")
 input_text = gr.inputs.Textbox(
     label="Input Text", lines=10, default="Enter your text here.")
 
+temperature_slider = gr.inputs.Slider(
+    label="Temperature", minimum=0, maximum=1, default=0.7, step=0.1)
+
 output_text = gr.outputs.Textbox(label="Output Text")
 
-io = gr.Interface(fn=replace_synonyms, inputs=[api_key, input_text], outputs=output_text, title="Synonym Replacer",
-                  description="Replace words in a text with their synonyms.", server_port=8080)
+io = gr.Interface(fn=replace_synonyms, inputs=[api_key, input_text, temperature_slider], outputs=output_text, title="Synonym Replacer",
+                  description="Replace words in a text with their synonyms.")
 io.launch()