complete

app.py

@@ -1,7 +1,32 @@
 import gradio as gr
 
-def greet(name):
-    return "Hello " + name + "!!"
+from fastcore.all import *
+from fastai.vision.all import *
+from fastai.vision.widgets import *
+import pickle
+import torch
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+
+def label_func (f):
+    return f[:2] !="1 "
+
+Pkl_Filename = "learner.pkl"
+with open(Pkl_Filename, 'rb') as file:  
+    learn = torch.load ( file , map_location=torch.device('cpu') )
+    
+category = ("Waifu" , "Human")
+def infer_im (im ) :
+    pre , idx , prob = learn.predict (im)
+    flo_prob = list ( map (float , prob) )[0]
+    return dict ( zip (category , ( flo_prob , 1 - flo_prob ) ) )
+
+
+examples = ["1 (1004).jpg",
+           "498970d2f45988eae9d6a2eb59bc450f.jpg" ,
+           "10010892_result.jpg" , 
+           "10019581_result.jpg"]
+
+gr.Interface(fn= infer_im, 
+             inputs=gr.inputs.Image(shape=(128 , 128)),
+             outputs=gr.outputs.Label(num_top_classes=2) ,
+            examples = examples ).launch(inline = False)

learner.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:92c7c6d0ee3a6fda84aa09011f0600388ec4569d34b91367bfa1c3b028ef8ad9
+size 141517491