Spaces:
Sleeping
Sleeping
Update detector/custom_model.py
Browse files- detector/custom_model.py +1 -0
detector/custom_model.py
CHANGED
|
@@ -6,6 +6,7 @@ from transformers import RobertaTokenizer, RobertaForSequenceClassification
|
|
| 6 |
from reportlab.lib.pagesizes import A4
|
| 7 |
from reportlab.pdfgen import canvas
|
| 8 |
from reportlab.lib import colors
|
|
|
|
| 9 |
import nltk
|
| 10 |
|
| 11 |
# === Environment Setup ===
|
|
|
|
| 6 |
from reportlab.lib.pagesizes import A4
|
| 7 |
from reportlab.pdfgen import canvas
|
| 8 |
from reportlab.lib import colors
|
| 9 |
+
from io import BytesIO
|
| 10 |
import nltk
|
| 11 |
|
| 12 |
# === Environment Setup ===
|