ai-forever commited on
Commit
54b2cb6
·
verified ·
1 Parent(s): e8662a1

Update src/constructor.py

Browse files
Files changed (1) hide show
  1. 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[:slides_num],
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")