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