Update main.py
Browse files
main.py
CHANGED
|
@@ -214,6 +214,6 @@ def load_checkpoint(filepath):
|
|
| 214 |
|
| 215 |
@app.route('/')
|
| 216 |
def home():
|
| 217 |
-
|
| 218 |
-
|
| 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}
|