Launch human or animal application with 2 models

app.py

@@ -1,9 +1,24 @@
-# Default hello world app.
+# Human or Animal application in gradio.
 
 import gradio as gr
+from fastai.vision.all import *
+import skimage
 
-def greet(name):
-    return "Hello " + name + "!!"
+loaded_generic_model = load_learner('model_generic.pkl')
+loaded_special_model = load_learner('model_special.pkl')
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+def label_func(filename):
+    if 'human' in Path(filename).parts[-2]:
+        return 'human'
+    elif 'animal' in Path(filename).parts[-2]:
+        return 'animal'
+
+def predict_human_or_animal(img):
+    im = PILImage.create(img)
+    generic_class, generic_class_id, generic_probs = loaded_generic_model.predict(img)
+    #special_class, special_class_id, special_probs = loaded_special_model.predict(img)
+    
+    return {generic_class: 1.0}
+    
+intf = gr.Interface(fn=predict_human_or_animal, inputs="image", outputs="label").launch()
+    

requirements.txt

@@ -0,0 +1,2 @@
+fastai
+scikit-image