Spaces:

TheOneReborn
/

car-segmentation

Sleeping

TheOneReborn commited on Mar 5

Commit

9ab897d

1 Parent(s): 4ae8579

fix: open file not directory

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ from fastai.vision.all import load_learner, PILImage
 MODEL_PATH = Path('.') / 'models'
 TEST_IMAGES_PATH = Path('.') / 'test'
-LEARNER = load_learner(MODEL_PATH) / 'car-segmentation_v1.pkl'
 def segment_image(image):

 MODEL_PATH = Path('.') / 'models'
 TEST_IMAGES_PATH = Path('.') / 'test'
+LEARNER = load_learner(MODEL_PATH / 'car-segmentation_v1.pkl')
 def segment_image(image):