AI_DEM / app.py
Nurisslam's picture
Update app.py
a24d772 verified
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)