|
|
from fastai.text.all import * |
|
|
import gradio as gr |
|
|
|
|
|
|
|
|
learn = load_learner('model.pkl') |
|
|
|
|
|
|
|
|
description = "Медициналық Диагностика" |
|
|
categories = ['Аллергия', 'Анемия', 'Бронхит', 'Диабет', 'Диарея', 'Шаршау', 'Тұмау', 'Малярия', 'Стресс'] |
|
|
|
|
|
|
|
|
def classify_text(txt): |
|
|
pred, idx, probs = learn.predict(txt) |
|
|
return dict(zip(categories, map(float, probs))) |
|
|
|
|
|
|
|
|
text = gr.Textbox(lines=2, label='Өзіңізді қалай сезінетініңізді егжей-тегжейлі сипаттаңыз') |
|
|
label = gr.Label() |
|
|
examples = [ |
|
|
'Менде физикалық белсенділікке қызығушылық жоқ. Мен әрқашан шөлдеп жүрмін', |
|
|
'Мен тоңып тұрмын', |
|
|
'Көзіңіз бозарған' |
|
|
] |
|
|
|
|
|
|
|
|
intf = gr.Interface(fn=classify_text, inputs=text, outputs=label, examples=examples, description=description) |
|
|
intf.launch(inline=False) |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|