raihanp commited on
Commit
7f16b2f
·
verified ·
1 Parent(s): d817bef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ inference = torch.load('fine_tune_resnet.pth', map_location=torch.device('cpu'))
15
  inference.eval()
16
 
17
 
18
- example = [Path.cwd() / 'cardboard.wp', Path.cwd() / 'plastic.jpg']
19
 
20
  def classifier(image):
21
  test_transform = Compose([
 
15
  inference.eval()
16
 
17
 
18
+ example = [str(i) for i in Path('examples').glob('*')]
19
 
20
  def classifier(image):
21
  test_transform = Compose([