Spaces:
Sleeping
Sleeping
Cherryblade29
commited on
Commit
•
0a0f565
1
Parent(s):
2c355da
Update app.py
Browse files
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(
|
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)
|