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() |