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}
|