Skip to content

Commit

Permalink
update predict() response
Browse files Browse the repository at this point in the history
  • Loading branch information
capjamesg committed Oct 27, 2023
1 parent 23cb10a commit beb4735
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions roboflow/models/video.py
Original file line number Diff line number Diff line change
Expand Up @@ -4,6 +4,7 @@

import requests
import time
from typing import List

from roboflow.config import API_URL
from roboflow.models.inference import InferenceModel
Expand Down Expand Up @@ -31,7 +32,7 @@ def __init__(
api_key,
):
"""
Create a ObjectDetectionModel object through which you can run inference.
Create a VideoDetectionModel object through which you can run inference on videos.
Args:
api_key (str): Your API key (obtained via your workspace API settings page).
Expand All @@ -41,7 +42,7 @@ def __init__(
def predict(
self,
video_path,
) -> str:
) -> List[str, str]:
"""
Infers detections based on image from specified model and image path.
Expand Down Expand Up @@ -106,7 +107,7 @@ def predict(

self.job_id = job_id

return job_id
return job_id, signed_url

def poll_for_results(self, job_id: str = None) -> dict:
"""
Expand Down

0 comments on commit beb4735

Please sign in to comment.