Update app.py

app.py

@@ -19,7 +19,8 @@ from fastai.vision.all import *
 # temp = pathlib.PosixPath
 # pathlib.PosixPath = pathlib.WindowsPath
 
-print(os.listdir())
+# Get the directory of the current script
+script_dir = os.path.dirname(os.path.realpath(__file__))
 
 st.set_page_config(page_title="Aplicación cancer de mama - IXCHEN")
 
@@ -31,9 +32,9 @@ class Modelo:
     
     def cargar_modelo(self, modelo):
         if modelo == "alexnet":
-            return load_learner('clasificacion_alexnet.pkl')
+            return load_learner(os.path.join(script_dir, 'clasificacion_alexnet.pkl'))
         elif modelo == 'restnet50':
-            return load_learner('clasificacion.pkl')
+            return load_learner(os.path.join(script_dir, 'clasificacion.pkl'))
         
     def predecir(self, imagen):
         _, _, probs = self.modelo.predict(imagen)