Update app.py

app.py

@@ -61,7 +61,7 @@ if btn:
                 theme,
                 topic,
             )
-            content.append(complete_with_gpt(prmpt, 200, "meta-llama/Llama-3.2-1B", 1500, 0.7, 1.5))
+            content.append(complete_with_gpt(prmpt, 200, "meta-llama/Llama-3.2-3B", 1500, 0.7, 1.5))
         if x == 1:
             prmpt = story_setting_prompt(
                 genre,
@@ -76,7 +76,7 @@ if btn:
             )
             previous = " ".join(x for x in content)
             prmpt = previous + " " + prmpt
-            content.append(complete_with_gpt(prmpt, 200, "meta-llama/Llama-3.2-1B", 1500, 0.7, 1.5))
+            content.append(complete_with_gpt(prmpt, 200, "meta-llama/Llama-3.2-3B", 1500, 0.7, 1.5))
 
         if x % 3 == 0:
             prmpt = supporting_character_inclusion(
@@ -87,16 +87,16 @@ if btn:
             )
             previous = " ".join(x for x in content)
             prmpt = previous + " " + prmpt
-            content.append(complete_with_gpt(prmpt, 200, "meta-llama/Llama-3.2-1B", 1500, 0.7, 1.5))
+            content.append(complete_with_gpt(prmpt, 200, "meta-llama/Llama-3.2-3B", 1500, 0.7, 1.5))
         if x == chapters - 1:
             prmpt = ending_scene(genre, resolution, main_name, positive_messaging)
             previous = " ".join(x for x in content)
             prmpt = previous + " " + prmpt
-            content.append(complete_with_gpt(prmpt, 200, "meta-llama/Llama-3.2-1B", 1500, 0.7, 1.5))
+            content.append(complete_with_gpt(prmpt, 200, "meta-llama/Llama-3.2-3B", 1500, 0.7, 1.5))
         else:
             previous = " ".join(x for x in content)
             prmpt = previous
-            content.append(complete_with_gpt(prmpt, 200, "meta-llama/Llama-3.2-1B", 1500, 0.7, 1.5))
+            content.append(complete_with_gpt(prmpt, 200, "meta-llama/Llama-3.2-3B", 1500, 0.7, 1.5))
 
     st.write(content)
     filenamee = to_pdf(convert(create_md(text=content, title=title)))