Upload 2 files

app.py

@@ -1,5 +1,6 @@
 from fastai.vision.all import *
 import gradio as gr
+import cv2
 
 learn = load_learner('resnet18_model.pkl')
 
@@ -11,12 +12,30 @@ def classify_image(image):
     is_people,_,probs = learn.predict(PILImage.create(image))
     return dict(zip(categories,map(float,probs)))
 
+def capture_image(cam):
+    cam = cv2.VideoCapture(0)
+    _, frame = cam.read()
+    frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+    cam.release()
+    return frame
 
-image = gr.inputs.Image(shape=(192,192))
+
+def webcam():
+    cam = cv2.VideoCapture(0)
+    while True:
+        _, frame = cam.read()
+        frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+        yield frame
+
+image = gr.inputs.Image(shape=(192, 192), label="Upload Image", source="upload")
+webcam = gr.inputs.Image(shape=(192, 192), label="Webcam Image", source="webcam")
 label = gr.outputs.Label()
+
 examples=['example1.jpg','example2.jpg','example3.jpg']
 
-intf =gr.Interface(fn=classify_image,inputs=image,outputs=label,examples=examples)
+intf = gr.Interface(fn=classify_image, inputs=[image, webcam], outputs=label, examples=examples,
+                    allow_flagging=False, capture_session=True)
+
 intf.launch(inline=False)
 
 

requirements.txt

@@ -1 +1,2 @@
-fastai
+fastai==1.0.61
+cv2