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