jalFaizy's picture
Upload app.py
eaeb709
raw
history blame
521 Bytes
import gradio as gr
from huggingface_hub import from_pretrained_keras
def infer(filepath):
image_tensor = np.load(filepath)
prediction = model.predict(np.expand_dims(image_tensor, axis=0))[0]
confidence = 100 * (1 - prediction[0])
return confidence
model = from_pretrained_keras('jalFaizy/3D_CNN')
filepath = gr.inputs.File()
text = gr.outputs.Textbox()
iface = gr.Interface(
infer,
filepath,
text,
title='3D CNN for CT scans',
examples=['example_1_normal.npy']
)
iface.launch()