Spaces:
Runtime error
Runtime error
EveSa
commited on
Update inference_t5.py
Browse files- src/inference_t5.py +0 -3
src/inference_t5.py
CHANGED
|
@@ -2,12 +2,9 @@
|
|
| 2 |
Allows to predict the summary for a given entry text
|
| 3 |
"""
|
| 4 |
import torch
|
| 5 |
-
import nltk
|
| 6 |
import contractions
|
| 7 |
import re
|
| 8 |
import string
|
| 9 |
-
nltk.download('stopwords')
|
| 10 |
-
nltk.download('punkt')
|
| 11 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
| 12 |
|
| 13 |
def clean_data(texts):
|
|
|
|
| 2 |
Allows to predict the summary for a given entry text
|
| 3 |
"""
|
| 4 |
import torch
|
|
|
|
| 5 |
import contractions
|
| 6 |
import re
|
| 7 |
import string
|
|
|
|
|
|
|
| 8 |
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
|
| 9 |
|
| 10 |
def clean_data(texts):
|