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']
|