Spaces:

Andreas-w
/

Brain-segmentation

Sleeping

Andreas-w commited on Apr 22, 2023

Commit

de77762

1 Parent(s): 62cbf53

Update metode.py

Files changed (1) hide show

metode.py CHANGED Viewed

@@ -17,7 +17,7 @@ def load_model(model_path):
     # Define the Learner object
     learn = Learner(dls, model, metrics=accuracy)
-return learn
 # Load the model from the .pth file and create the necessary objects
 #learn = load_model('my_model.pth')

     # Define the Learner object
     learn = Learner(dls, model, metrics=accuracy)
+    return learn
 # Load the model from the .pth file and create the necessary objects
 #learn = load_model('my_model.pth')