replacing new model

app.py

@@ -1,25 +1,15 @@
-from fastai.vision.all import *
-from fastcore.all import *
-import pickle
+from fastai.vision.all import load_learner, get_image_files
 
-learn = load_learner(
-    'resnet18_bearclassifier.pkl',
-    pickle_module={
-        'pickle_module': pickle,
-        'map_location': lambda storage, loc: storage
-    }
-)
-
-# gradio UI
+learn = load_learner('resnet18_bearclassifier.pkl')
 
+#gradio UI
+import gradio as gr
 
 def classify_image(img):
     categories = ['brown', 'polar']
-    _, _, probability = learn.predict(img)
+    _,_, probability = learn.predict(img)
     return dict(zip(categories, map(float, probability)))
 
-
-app = gr.Interface(fn=classify_image, inputs='image',
-                   outputs='label', examples=list(get_image_files('valid')))
+app = gr.Interface(fn=classify_image, inputs='image', outputs='label', examples= list(get_image_files('valid')))
 app.launch()
 classify_image('images.jpg')

resnet18_bearclassifier.pkl → resnet18_bearclassifier_lunux.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:00bce8ac5b8c770c019669f9a92fea24b99c1742d0b8d3eeae852d898040cc01
-size 46965914
+oid sha256:a393751fe74333b62f1c300803d388df75d23f1e317f69d716cfe6f42397dabd
+size 46967194