1qwsd commited on
Commit
9628d2a
·
1 Parent(s): eadea58

fix: use standard python 3.11 base image and reorder torch imports

Browse files
Files changed (2) hide show
  1. Dockerfile +1 -1
  2. models/owl_vit_detector.py +1 -2
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- FROM python:3.11-slim
2
 
3
  # Install system deps for OpenCV, ffmpeg, git (needed for CLIP install)
4
  RUN apt-get update && apt-get install -y \
 
1
+ FROM python:3.11
2
 
3
  # Install system deps for OpenCV, ffmpeg, git (needed for CLIP install)
4
  RUN apt-get update && apt-get install -y \
models/owl_vit_detector.py CHANGED
@@ -1,7 +1,6 @@
1
- """OWL-ViT Object Detection"""
2
  from transformers import pipeline
3
  from core.config import settings
4
- import torch
5
 
6
  class OWLViTDetector:
7
  def __init__(self):
 
1
+ import torch
2
  from transformers import pipeline
3
  from core.config import settings
 
4
 
5
  class OWLViTDetector:
6
  def __init__(self):