DoctorSlimm commited on
Commit
9d4f0df
·
verified ·
1 Parent(s): e25568a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -33,9 +33,10 @@ def greet(prompts, separator):
33
  sampling_params = SamplingParams(**sampling_params)
34
 
35
  multi_prompt = False
 
36
  if separator in prompts:
37
  multi_prompt = True
38
- prompts = prompts.split('separator')
39
  else:
40
  prompts = [prompts]
41
  for idx, pt in enumerate(prompts):
 
33
  sampling_params = SamplingParams(**sampling_params)
34
 
35
  multi_prompt = False
36
+ separator = separator.strip()
37
  if separator in prompts:
38
  multi_prompt = True
39
+ prompts = prompts.split(separator)
40
  else:
41
  prompts = [prompts]
42
  for idx, pt in enumerate(prompts):