Upload app.py

app.py

@@ -1,18 +1,26 @@
+# AUTOGENERATED! DO NOT EDIT! File to edit: ../app.ipynb.
+
+# %% auto 0
+__all__ = ['learn', 'labels', 'inputs', 'label', 'intf', 'classify_image']
+
+# %% ../app.ipynb 1
+import numpy as np
 from fastai.vision.all import *
 import gradio as gr
 
-# model
+# %% ../app.ipynb 3
 learn = load_learner("face_mask_detection_model.pkl")
-labels = {"Wearing Mask", "No Mask"}
+
+# %% ../app.ipynb 5
+labels = ["Wearing Mask", "No Mask"]
 
 def classify_image(img):
     pred, idx, probs = learn.predict(img)
-    return dict(zip(labels, map(float, probs)))
+    return labels[np.argmax(probs)]
 
-# gradio interface
-image = gr.inputs.Image(shape=(224,224))
+# %% ../app.ipynb 6
+inputs = gr.Webcam(mirror_webcam=True, streaming=True)
 label = gr.outputs.Label()
-examples = ["example.jpg"]
 
-intf = gr.Interface(fn=classify_image, inputs=image, outputs=label, examples=examples)
-intf.launch(inline=False)
+intf = gr.Interface(fn=classify_image, inputs=inputs, outputs=label)
+intf.launch(inline=False, live=True)