Update ai_generator.py
Browse files- ai_generator.py +1 -1
ai_generator.py
CHANGED
@@ -12,7 +12,7 @@ def generate_content(topic: str):
|
|
12 |
prompt = f"Create a detailed and engaging YouTube script about: {topic}"
|
13 |
|
14 |
# Generate content
|
15 |
-
generated_text = generator(prompt, max_length=
|
16 |
|
17 |
# Extract the generated text (it's a list of dicts)
|
18 |
return generated_text[0]['generated_text']
|
|
|
12 |
prompt = f"Create a detailed and engaging YouTube script about: {topic}"
|
13 |
|
14 |
# Generate content
|
15 |
+
generated_text = generator(prompt, max_length=2000, num_return_sequences=1, temperature=0.7, top_p=0.9)
|
16 |
|
17 |
# Extract the generated text (it's a list of dicts)
|
18 |
return generated_text[0]['generated_text']
|