ANASAKHTAR commited on
Commit
c9d9274
·
verified ·
1 Parent(s): 4d409f3

Update ai_generator.py

Browse files
Files changed (1) hide show
  1. 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=500, 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']
 
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']