merge_test / app.py
SaintJagh
Add model and app files
a91f00e
import gradio as gr
import torch
from transformers import pipeline
# بارگذاری مدل
model = pipeline("text-classification", model="rinrikatoki/dorna-merged-diabetes")
def predict(text):
result = model(text)
return result
# ساخت اینترفیس Gradio
iface = gr.Interface(
fn=predict,
inputs="text",
outputs="label",
title="Diabetes Prediction Model",
description="A demo for diabetes classification model."
)
iface.launch()