| import gradio as gr | |
| from transformers import pipeline | |
| import datetime | |
| generator = pipeline('text2text-generation', model='mohsenfayyaz/mt5-small-query_realestate_cars-finetuned') | |
| def generate(text): | |
| print(f"[{datetime.datetime.now().strftime('%Y-%m-%d %H:%M')}]: {text}") | |
| result = generator(text, max_length=200) | |
| return result[0]["generated_text"] | |
| examples = [ | |
| ["آپارتمان 100 متری توی پیروزی سه خواب"], | |
| ["اجاره آپارتمان ۶۰ متر صادقیه"], | |
| ["شمع 206"], | |
| ["ماشین پژو پارس سفید مدل 90"], | |
| ] | |
| demo = gr.Interface( | |
| fn=generate, | |
| inputs=gr.inputs.Textbox(lines=1, label="Input Text"), | |
| outputs=gr.outputs.Textbox(label="Generated Text"), | |
| examples=examples, | |
| cache_examples=True, | |
| title="Query Information Extraction Demo", | |
| ) | |
| demo.launch() |