DoctorSlimm commited on
Commit
42814c6
·
verified ·
1 Parent(s): 27cf068

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -40,6 +40,8 @@ def greet(prompts, separator):
40
  if separator in prompts:
41
  multi_prompt = True
42
  prompts = prompts.split('separator')
 
 
43
  model_outputs = model.generate(prompts, sampling_params)
44
  generations = []
45
  for output in model_outputs:
 
40
  if separator in prompts:
41
  multi_prompt = True
42
  prompts = prompts.split('separator')
43
+ else:
44
+ prompts = [prompts]
45
  model_outputs = model.generate(prompts, sampling_params)
46
  generations = []
47
  for output in model_outputs: