File size: 1,175 Bytes
112f14c b502ad3 112f14c b502ad3 112f14c |
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 |
import gradio as gr
import os
import sys
from generate import main
def generate_text(length, nsamples, prefix, temperature):
my_prefix = "--prefix=" + prefix + ","
sys.argv = [
'generate.py',
f'--length={length}',
f'--nsamples={nsamples}',
my_prefix,
f'--temperature={temperature}',
'--model_path=model/model_epoch10'
]
output = main()
return output
input_length = gr.inputs.Number(label="ηζζζ¬ιΏεΊ¦", min_value=1, max_value=10000, default=1000)
input_nsamples = gr.inputs.Number(label="ηζζζ¬ζ°ι", min_value=1, max_value=10, default=3)
input_prefix = gr.inputs.Textbox(label="θ΅·ε§ζζ¬")
input_temperature = gr.inputs.Slider(label="ηζζΈ©εΊ¦", minimum=0.1, maximum=2.0, default=0.7, step=0.1)
output_text = gr.outputs.Textbox(label="ηζηζζ¬")
title = "GPT2δΈζζζ¬ηζε¨"
description = "θΎε
₯θ΅·ε§ζζ¬γηζζζ¬ιΏεΊ¦γηζζζ¬ζ°ιεηζζΈ©εΊ¦οΌηζδΈζζζ¬γ"
gr.Interface(fn=generate_text, inputs=[input_length, input_nsamples, input_prefix, input_temperature], outputs=output_text, title=title, description=description).launch()
|