Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
·
81b7588
1
Parent(s):
a864526
Update predict_model.py
Browse files
src/models/predict_model.py
CHANGED
@@ -10,6 +10,7 @@ def predict_model(text):
|
|
10 |
pre_summary = model.predict(text)
|
11 |
return pre_summary
|
12 |
|
|
|
13 |
if __name__ == '__main__':
|
14 |
text = make_dataset(split="test")['input_text'][0]
|
15 |
pre_summary = predict_model(text)
|
|
|
10 |
pre_summary = model.predict(text)
|
11 |
return pre_summary
|
12 |
|
13 |
+
|
14 |
if __name__ == '__main__':
|
15 |
text = make_dataset(split="test")['input_text'][0]
|
16 |
pre_summary = predict_model(text)
|