Cherryblade29 commited on
Commit
0a0f565
1 Parent(s): 2c355da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,7 +16,7 @@ effnetb2 , effnetb2_transforms = create_effnetb2_model(num_classes=3)
16
 
17
  # load and save weights
18
 
19
- effnetb2.load_state_dict(torch.load(os.path.join("effnetb2.pth"),map_location=torch.device('cpu')))
20
 
21
  #effnetb2.load_state_dict(torch.load("effnetb2.pth",map_location=torch.device('cpu')))
22
 
@@ -66,4 +66,4 @@ demo = gr.Interface(fn=predict, # mapping function from input to output
66
  article=article)
67
 
68
  # Launch the demo!
69
- demo.launch()
 
16
 
17
  # load and save weights
18
 
19
+ effnetb2.load_state_dict(torch.load("effnetb2.pth",map_location=torch.device('cpu')))
20
 
21
  #effnetb2.load_state_dict(torch.load("effnetb2.pth",map_location=torch.device('cpu')))
22
 
 
66
  article=article)
67
 
68
  # Launch the demo!
69
+ demo.launch(debug=True)