Danil commited on
Commit
2525f21
·
1 Parent(s): 09347b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -7,7 +7,11 @@ openai.api_key = os.environ["api_key"]
7
  engine = os.environ["engine"]
8
 
9
 
10
- def happytt(temperature,max_tokens,text,stop):
 
 
 
 
11
  response = openai.Completion.create(
12
  engine=engine,
13
  prompt=text,
@@ -16,7 +20,7 @@ def happytt(temperature,max_tokens,text,stop):
16
  top_p=1,
17
  frequency_penalty=0,
18
  presence_penalty=0,
19
- stop=[stop]
20
  )
21
 
22
  return response.choices[0].text
 
7
  engine = os.environ["engine"]
8
 
9
 
10
+ def happytt(temperature,max_tokens,text,stop):
11
+ if stop == '':
12
+ stop = []
13
+ else:
14
+ stop = [stop]
15
  response = openai.Completion.create(
16
  engine=engine,
17
  prompt=text,
 
20
  top_p=1,
21
  frequency_penalty=0,
22
  presence_penalty=0,
23
+ stop=stop
24
  )
25
 
26
  return response.choices[0].text