Spaces:
Sleeping
Sleeping
Update predictions.py
Browse files- predictions.py +7 -1
predictions.py
CHANGED
|
@@ -1,6 +1,12 @@
|
|
| 1 |
from PIL import Image, ImageDraw
|
| 2 |
-
from helper import summarize_predictions_natural_language,render_results_in_image
|
| 3 |
from transformers import pipeline
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 4 |
|
| 5 |
# Load object detection pipeline
|
| 6 |
object_detection_pipe = pipeline("object-detection", model="facebook/detr-resnet-50")
|
|
|
|
| 1 |
from PIL import Image, ImageDraw
|
| 2 |
+
from helper import summarize_predictions_natural_language, render_results_in_image
|
| 3 |
from transformers import pipeline
|
| 4 |
+
from tokenizers import Tokenizer, Encoding
|
| 5 |
+
from tokenizers import decoders
|
| 6 |
+
from tokenizers import models
|
| 7 |
+
from tokenizers import normalizers
|
| 8 |
+
from tokenizers import pre_tokenizers
|
| 9 |
+
from tokenizers import processors
|
| 10 |
|
| 11 |
# Load object detection pipeline
|
| 12 |
object_detection_pipe = pipeline("object-detection", model="facebook/detr-resnet-50")
|