Spaces:
Sleeping
Sleeping
fix: use standard python 3.11 base image and reorder torch imports
Browse files- Dockerfile +1 -1
- models/owl_vit_detector.py +1 -2
Dockerfile
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 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 \
|
|
|
|
| 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 |
-
|
| 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):
|