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

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +6 -3
main.py CHANGED
@@ -214,6 +214,9 @@ 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
+ print(1)
218
+ model = load_checkpoint('checkpoint.pth')
219
+ print(2)
220
+ res = sample(model, obj.maxlen, 'ap')
221
+ print(3)
222
+ return {'key':res}