Upload app.py

app.py

@@ -7,9 +7,13 @@ learn = load_learner('resnet18_model.pkl')
 categories = ('Black', 'White')
 
 
-def classify_image(image):
-    is_people, _, probs = learn.predict(PILImage.create(image))
-    return dict(zip(categories, map(float, probs)))
+def classify_image(image, webcam=None):
+    if image is not None:
+        is_people, _, probs = learn.predict(PILImage.create(image))
+        return dict(zip(categories, map(float, probs)))
+    elif webcam is not None:
+        is_people, _, probs = learn.predict(PILImage.create(webcam))
+        return dict(zip(categories, map(float, probs)))
 
 
 def capture_image(cam):