Update app.py

app.py

@@ -9,29 +9,25 @@ import gradio as gr
 import pathlib
 import platform
 
-
 def whichBear(x): return x[0].isBear()
 
 # %% app.ipynb 4
-original_posix_path = pathlib.PosixPath
+# Function to handle the path conversion if necessary
+def fix_path():
+    if platform.system().lower() != "windows":
+        pathlib.WindowsPath = pathlib.PosixPath
 
-# Replace PosixPath with WindowsPath for loading the model if on a POSIX system
-if platform.system().lower() != "windows":
-    pathlib.PosixPath = pathlib.WindowsPath
+# Call the function to fix path handling before loading the model
+fix_path()
 
 # Load the learner
 learn = load_learner("model.pkl")
 
-# Restore the original PosixPath class
-if platform.system().lower() != "windows":
-    pathlib.PosixPath = original_posix_path
-
-
 # %% app.ipynb 6
 options = ("grizzly", "black", "teddy")
 
 def classify_img(img):
-    pred,idx,probs = learn.predict(img)
+    pred, idx, probs = learn.predict(img)
     return dict(zip(options, map(float, probs)))
 
 # %% app.ipynb 8