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")
|