Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
·
972bccc
1
Parent(s):
7f8704f
Update predict_model.py
Browse files
src/models/predict_model.py
CHANGED
|
@@ -9,6 +9,8 @@ def predict_model(text):
|
|
| 9 |
Predict the summary of the given text.
|
| 10 |
"""
|
| 11 |
with open("params.yml") as f:
|
|
|
|
|
|
|
| 12 |
|
| 13 |
model = Summarization()
|
| 14 |
model.load_model()
|
|
|
|
| 9 |
Predict the summary of the given text.
|
| 10 |
"""
|
| 11 |
with open("params.yml") as f:
|
| 12 |
+
params = yaml.safe_load(f)
|
| 13 |
+
|
| 14 |
|
| 15 |
model = Summarization()
|
| 16 |
model.load_model()
|