Hugging Face's logo

Spaces:
gagan3012
/
T5-Summarization
Runtime error

App Files Community
Gagan Bhatia commited on
Commit
f19d75b
•
1 Parent(s): df90151

Update predict_model.py

Browse files
Files changed (1) hide show
  1. src/models/predict_model.py +2 -1
src/models/predict_model.py CHANGED
@@ -1,4 +1,5 @@
1
- from src.data.make_dataset import make_dataset
 
2
  from .model import Summarization
3
 
4
  def predict_model(text):
 
1
+ import yaml
2
+
3
  from .model import Summarization
4
 
5
  def predict_model(text):