kumar9 commited on
Commit
63c5af1
1 Parent(s): 67c57d4

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -2
main.py CHANGED
@@ -214,6 +214,6 @@ def load_checkpoint(filepath):
214
 
215
  @app.route('/')
216
  def home():
217
- model = load_checkpoint('checkpoint.pth')
218
- res = sample(model, obj.maxlen, 'ap')
219
  return {'key':res}
 
214
 
215
  @app.route('/')
216
  def home():
217
+ model = load_checkpoint('checkpoint.pth')
218
+ res = sample(model, obj.maxlen, 'ap')
219
  return {'key':res}