initial commit

laod_pipeline.py

@@ -3,6 +3,7 @@ from transformers import pipeline
 from transformers import AutoProcessor, AutoModelForZeroShotObjectDetection
 import numpy as np
 import draw_utils
+import spaces
 
 from huggingface_hub import login
 import os
@@ -30,6 +31,7 @@ model_id = "IDEA-Research/grounding-dino-tiny"
 processor = AutoProcessor.from_pretrained(model_id)
 model = AutoModelForZeroShotObjectDetection.from_pretrained(model_id).to(device)
 
+@spaces.GPU
 def laod_gdino(image):
     messages = [
             {

requirements.txt

@@ -7,3 +7,5 @@ Pillow
 opencv-python
 huggingface-hub
 accelerate
+spaces
+