minar09 commited on
Commit
a5108b5
·
verified ·
1 Parent(s): e9bce2a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -12,6 +12,7 @@ from dds_cloudapi_sdk.tasks.types import DetectionTarget
12
  # Constants
13
  API_TOKEN = "361d32fa5ce22649133660c65cfcaf22"
14
  TEXT_PROMPT = "wheel . eye . helmet . mouse . mouth . vehicle . steering wheel . ear . nose"
 
15
  TEMP_DIR = "./temp"
16
  OUTPUT_DIR = "./outputs"
17
 
@@ -116,7 +117,7 @@ def process_image(image_path, prompt=TEXT_PROMPT):
116
  except Exception as e:
117
  return f"Error processing image: {str(e)}"
118
 
119
- def process_video(video_path, prompt=TEXT_PROMPT):
120
  """Process a video with DINO-X"""
121
  try:
122
  client = initialize_dino_client()
 
12
  # Constants
13
  API_TOKEN = "361d32fa5ce22649133660c65cfcaf22"
14
  TEXT_PROMPT = "wheel . eye . helmet . mouse . mouth . vehicle . steering wheel . ear . nose"
15
+ VID_PROMPT = "wheel . mouse . pot . acquariam . box"
16
  TEMP_DIR = "./temp"
17
  OUTPUT_DIR = "./outputs"
18
 
 
117
  except Exception as e:
118
  return f"Error processing image: {str(e)}"
119
 
120
+ def process_video(video_path, prompt=VID_PROMPT):
121
  """Process a video with DINO-X"""
122
  try:
123
  client = initialize_dino_client()