Spaces:

fsattar
/

brain-tumor-classification

Sleeping

fsattar commited on 15 days ago

Commit

6556a59

verified ·

1 Parent(s): 6029141

change name of model loaded

Files changed (1) hide show

brain_tumor_resnet34.ipynb CHANGED Viewed

@@ -728,7 +728,7 @@
       ],
       "source": [
         "# load the model back into learn\n",
-        "learn = load_learner('brain.pkl')\n",
         "print(f'Reloaded vocab: {learn.dls.vocab}')"
       ]
     },

       ],
       "source": [
         "# load the model back into learn\n",
+        "learn = load_learner('brain_tumor_resnet34.pkl')\n",
         "print(f'Reloaded vocab: {learn.dls.vocab}')"
       ]
     },