working good now

app.py

@@ -9,21 +9,19 @@ def label_func(x):
     else:
         return 'Mogged'
     
-learn = load_learner('model.pkl')
+learn = load_learner('mommodel.pkl')
 
-# categories = {'Mogger', 'Mogged'}
+categories = {'Mogger', 'Mogged'}
 
 def classify_img(image):
-    what, _t , _ = learn.predict(image)
-    return what
-    # pred, idx, probs = learn.predict(image)
-    # return dict(zip(categories, map(float, probs)))
+    # what, _t , _ = learn.predict(image)
+    # return what
+    pred, idx, probs = learn.predict(image)
+    return dict(zip(categories, map(float, probs)))
 
-image = gr.Image()
+image = gr.Image(shape=(224,224))
 label = gr.Label()
 examples = ['001.jpg', '002.jpg', '007.jpg', '008.jpg']
 
-
-
 intf = gr.Interface(fn=classify_img, inputs=image, outputs=label, examples=examples)
 intf.launch(inline=False)

model.pkl → mommodel.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:429bd2f8638d8357a4098a31cc571a98dc777c35f0db367c597ff262d5ee3436
-size 179162534
+oid sha256:830f8b69a882cf030ae81d67e45cf66ab7040a4e0a79fc7b4aa5fe6d1d665704
+size 46962605