Spaces:
Runtime error
Runtime error
File size: 2,505 Bytes
d31c634 c53fa23 d31c634 b900928 c53fa23 afad1bb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 |
import gradio as gr
from humanize import paraphrase_text
with gr.Blocks() as demo:
gr.Markdown("# Polygraf Writer")
with gr.Row():
with gr.Column(scale=0.7):
gr.Markdown("## Enter a topic to write an article about:")
input_topic = gr.Textbox(label="Topic")
gr.Markdown("### Humanized article:")
output_label = gr.HTML(label="Output")
model_dropdown = gr.Radio(
choices=[
"Base Model",
"Large Model",
"XL Model",
"XL Law Model",
"XL Marketing Model",
"XL Child Style Model",
],
value="Large Model",
label="Select Model Version",
)
process_button = gr.Button("Humanize Text")
with gr.Column(scale=0.3):
temperature_slider = gr.Slider(minimum=0.5, maximum=2.0, step=0.1, value=1.2, label="Temperature")
gr.Markdown("Controls the randomness of the paraphrase. Higher values generate more varied text.")
top_k_slider = gr.Slider(
minimum=0,
maximum=300,
step=25,
value=50,
label="Top k",
)
gr.Markdown("Limits the number of top tokens considered during generation.")
repetition_penalty_slider = gr.Slider(
minimum=1.0,
maximum=2.0,
step=0.1,
value=1,
label="Repetition Penalty",
)
gr.Markdown("Penalizes repeated words to encourage diverse language use")
length_penalty_slider = gr.Slider(
minimum=0.0,
maximum=2.0,
step=0.1,
value=1.0,
label="Length Penalty",
)
gr.Markdown("Penalizes shorter outputs.")
process_button.click(
fn=paraphrase_text,
inputs=[
input_text,
model_dropdown,
temperature_slider,
repetition_penalty_slider,
top_k_slider,
length_penalty_slider,
],
outputs=output_label,
)
if __name__ == "__main__":
demo.launch(demo.launch(server_name="0.0.0.0"))
|