Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
0c204b8
1
Parent(s):
241e6ac
Update predict_model.py
Browse files
src/models/predict_model.py
CHANGED
@@ -17,7 +17,7 @@ def predict_model(text):
|
|
17 |
pre_summary = model.predict(text)
|
18 |
return pre_summary
|
19 |
|
20 |
-
|
21 |
if __name__ == '__main__':
|
22 |
text = make_dataset(split="test")['input_text'][0]
|
23 |
pre_summary = predict_model(text)
|
|
|
17 |
pre_summary = model.predict(text)
|
18 |
return pre_summary
|
19 |
|
20 |
+
|
21 |
if __name__ == '__main__':
|
22 |
text = make_dataset(split="test")['input_text'][0]
|
23 |
pre_summary = predict_model(text)
|