Update app.py
Browse files
app.py
CHANGED
|
@@ -11,7 +11,15 @@ model = GPT2LMHeadModel.from_pretrained(model_name)
|
|
| 11 |
def generate_blogpost(topic):
|
| 12 |
prompt = f"Write a detailed blog post about {topic}."
|
| 13 |
inputs = tokenizer.encode(prompt, return_tensors="pt")
|
| 14 |
-
outputs = model.generate(
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 15 |
generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
| 16 |
return generated_text
|
| 17 |
|
|
|
|
| 11 |
def generate_blogpost(topic):
|
| 12 |
prompt = f"Write a detailed blog post about {topic}."
|
| 13 |
inputs = tokenizer.encode(prompt, return_tensors="pt")
|
| 14 |
+
outputs = model.generate(
|
| 15 |
+
inputs,
|
| 16 |
+
max_length=300,
|
| 17 |
+
num_return_sequences=1,
|
| 18 |
+
pad_token_id=tokenizer.eos_token_id,
|
| 19 |
+
do_sample=True, # Enable sampling
|
| 20 |
+
temperature=0.7, # Control the randomness of the output
|
| 21 |
+
top_p=0.9 # Nucleus sampling
|
| 22 |
+
)
|
| 23 |
generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
| 24 |
return generated_text
|
| 25 |
|