chethu commited on
Commit
85a5dbd
1 Parent(s): a265c6d

Update predictions.py

Browse files
Files changed (1) hide show
  1. 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")