AI_novel / app.py
soJaeyoon's picture
Update app.py
fc855e4 verified
Raw
History Blame Contribute Delete
1.25 kB
import gradio as gr
from transformers import GPT2LMHeadModel, GPT2Tokenizer
# Load pre-trained GPT-2 model and tokenizer
tokenizer = GPT2Tokenizer.from_pretrained("gpt2")
model = GPT2LMHeadModel.from_pretrained("gpt2")
def generate_opposing_article(input_text, max_length=100):
# Encode input text
input_ids = tokenizer.encode(input_text, return_tensors="pt")
# Generate text with an opposing viewpoint
output = model.generate(
input_ids=input_ids,
max_length=max_length,
num_return_sequences=1,
pad_token_id=tokenizer.eos_token_id,
temperature=0.7,
top_k=50,
top_p=0.92,
repetition_penalty=1.0,
do_sample=True
)
# Decode generated text
generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
return generated_text
# Create Gradio interface
inputs = gr.inputs.Textbox(lines=5, label="Enter your statement:")
outputs = gr.outputs.Textbox(label="Opposing Article:")
gr.Interface(
fn=generate_opposing_article,
inputs=inputs,
outputs=outputs,
title="Opposing Article Generator",
description="Enter a statement, and this tool will generate an article with an opposing viewpoint.",
).launch()