Spaces:
Running
on
Zero
Running
on
Zero
Update src/constructor.py
Browse files- src/constructor.py +4 -4
src/constructor.py
CHANGED
@@ -49,21 +49,21 @@ def generate_presentation(
|
|
49 |
pbar = tqdm.tqdm(total=4, desc="Presentation goes brrr...")
|
50 |
|
51 |
pbar.set_description("Generating titles for presentation")
|
52 |
-
titles = llm_generate_titles(llm_generate, description + f", {slides_num} slides", prompt_config)
|
53 |
pbar.update(1)
|
54 |
|
55 |
pbar.set_description("Generating text for slides")
|
56 |
texts = [None] + llm_generate_text(
|
57 |
llm_generate,
|
58 |
description + f", {slides_num} slides",
|
59 |
-
titles
|
60 |
prompt_config
|
61 |
-
)
|
62 |
pbar.update(1)
|
63 |
|
64 |
# postfix added to keywords describing presentation
|
65 |
background_style = random.choice(prompt_config.background_styles)
|
66 |
-
|
67 |
picture_paths = []
|
68 |
background_paths = []
|
69 |
pbar.set_description("Generating images for slides")
|
|
|
49 |
pbar = tqdm.tqdm(total=4, desc="Presentation goes brrr...")
|
50 |
|
51 |
pbar.set_description("Generating titles for presentation")
|
52 |
+
titles = llm_generate_titles(llm_generate, description + f", {slides_num} slides", prompt_config)[:slides_num]
|
53 |
pbar.update(1)
|
54 |
|
55 |
pbar.set_description("Generating text for slides")
|
56 |
texts = [None] + llm_generate_text(
|
57 |
llm_generate,
|
58 |
description + f", {slides_num} slides",
|
59 |
+
titles,
|
60 |
prompt_config
|
61 |
+
)[:slides_num]
|
62 |
pbar.update(1)
|
63 |
|
64 |
# postfix added to keywords describing presentation
|
65 |
background_style = random.choice(prompt_config.background_styles)
|
66 |
+
|
67 |
picture_paths = []
|
68 |
background_paths = []
|
69 |
pbar.set_description("Generating images for slides")
|