Add token that indicate the end of the generation

app.py

@@ -7,7 +7,7 @@ openai.api_key = os.environ["api_key"]
 engine = os.environ["engine"]
 
 
-def happytt(temperature,max_tokens,text): 
+def happytt(temperature,max_tokens,text,stop): 
 
   response = openai.Completion.create(
     engine=engine,
@@ -16,7 +16,8 @@ def happytt(temperature,max_tokens,text):
     max_tokens=max_tokens,
     top_p=1,
     frequency_penalty=0,
-    presence_penalty=0
+    presence_penalty=0,
+    stop=[stop]
   )
 
   return response.choices[0].text
@@ -39,5 +40,7 @@ https://beta.openai.com/examples?category=code'''
  
 iface = gr.Interface( happytt,[ gr.inputs.Slider(0, 1, step=0.1),gr.inputs.Slider(150, 4000, step=1),
                                gr.inputs.Textbox(type='str',   
-                                                 label="input prompt")],"text", title = title, description = description )
+                                                 label="input prompt"),
+                               gr.inputs.Textbox(type='str', 
+                                                 label="Token that indicate the end of the generation")],"text", title = title, description = description )
 iface.launch(debug=True)