MaskDetector / app.py
SalmanHabeeb's picture
Update app.py
510708c
raw
history blame contribute delete
376 Bytes
import gradio as gr
import os
from fastai.vision.all import load_learner
model = load_learner("model.pkl")
def predict(img):
prediction, idx, confidence = model.predict(img)
return prediction
image = gr.inputs.Image(shape=(200, 200), source="webcam")
label = gr.outputs.Label()
interface = gr.Interface(fn=predict, inputs=image, outputs=label)
interface.launch()