wb-droid commited on
Commit
b43a417
1 Parent(s): c45e9cd

Use torch save.

Browse files
Files changed (2) hide show
  1. app.py +3 -2
  2. model_v4t.pkl +3 -0
app.py CHANGED
@@ -108,8 +108,9 @@ 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
 
114
  top_k = 20
115
  def generate(s, num = 60):
 
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
116
  def generate(s, num = 60):
model_v4t.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9e56c8c0d3d1fb4c25547214ed8a32f2e22fb71a1273cbd6fba4ab2661e2a319
3
+ size 4132106