Update app.py

app.py

@@ -1,5 +1,6 @@
 from fastai.vision.all import *
 import gradio as gr
+import cv2
 
 __all__ = ['is_rock', 'learn', 'classify_image', 'categories', 'image', 'label', 'examples', 'intf'] 
 
@@ -14,16 +15,16 @@ def classify_images(img):
     pred, idx, probs = learn.predict(img)
     return dict(zip(categories, map(float, probs)))
 
-#def get_webcam_image():
-#    cap = cv2.VideoCapture(0)
-#    _, frame = cap.read()
-#    cap.release()
-#    return frame
+def get_webcam_image():
+    cap = cv2.VideoCapture(0)
+    _, frame = cap.read()
+    cap.release()
+    return frame
 
 
 image = gr.inputs.Image(shape=(192,192))
 label = gr.outputs.Label()
 examples = ['rock.jpg', 'paper.jpg', 'scissor.jpg']
 
-intf = gr.Interface(fn = classify_images, inputs = webcam_image, outputs = label, examples = examples)
+intf = gr.Interface(fn = classify_images, inputs = get_webcam_imageimage, outputs = label, examples = examples)
 intf.launch(inline=False)