Use torch save.
Browse files
app.py
CHANGED
|
@@ -108,8 +108,6 @@ class Model(nn.Module):
|
|
| 108 |
m = Model()
|
| 109 |
m.to(device)
|
| 110 |
|
| 111 |
-
#with open("model_v4.pkl","rb") as f:
|
| 112 |
-
#m=pickle.load(f)
|
| 113 |
m=torch.load("model_v4t.pkl",map_location=torch.device('cpu'))
|
| 114 |
|
| 115 |
top_k = 20
|
|
|
|
| 108 |
m = Model()
|
| 109 |
m.to(device)
|
| 110 |
|
|
|
|
|
|
|
| 111 |
m=torch.load("model_v4t.pkl",map_location=torch.device('cpu'))
|
| 112 |
|
| 113 |
top_k = 20
|