Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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:
|