Create app.py

app.py

@@ -0,0 +1,27 @@
+from fastbook import *
+from fastai.vision.all import *
+import gradio as gr
+
+import re
+
+def classify_gender(path):
+    pattern = r"^man"
+    match = re.search(pattern, Path(path).name)
+    if match is not None and match.group() is not None:
+        if "man" in match.group():
+            return "man"
+    return "woman"
+
+learn = load_learner("model.pkl")
+
+genders = ("man", "woman")
+
+def classify_image(img):
+    pred, idx, probs = learn.predict(img)
+    return dict(zip(genders, map(float, probs)))
+
+image = gr.Image()
+label = gr.Label()
+
+intf = gr.Interface(fn = classify_image, inputs = image, outputs = label)
+intf.launch(inline = False, share = True)