Spaces:
Runtime error
Runtime error
import gradio as gr | |
import random | |
import markovify | |
def gen(text): | |
text = text.lower() | |
size = random.randint(1, 650) | |
file_name = "./dataset.txt" | |
with open(file_name, "a", encoding="utf-8") as f: | |
f.write(f"{text}\n") | |
with open(file_name, encoding="utf-8") as f: | |
db = f.read() | |
db = db.strip().lower() | |
text_model = markovify.NewlineText(input_text=db, state_size=1, well_formed=False) | |
symbolsplus = [".","~~","!","?"] | |
sentence: str = text_model.make_short_sentence(size) or random.choice( | |
db.splitlines())[:size] + random.choice(symbolsplus) | |
return sentence | |
iface = gr.Interface( | |
fn=gen, | |
inputs="text", | |
outputs="text", | |
examples=[["Привет!"], ["Как дела?"]]) | |
iface.launch() |