Spaces:

build-small-hackathon
/

tiny-trigger

Sleeping

tiny-trigger / tiny_trigger /__init__.py

Javier Montalvo

Added more providers, changed entrypoint, removed local llama for now

2a9b8d1 4 months ago

845 Bytes

"""Open-vocabulary video automations for Tiny Trigger."""

from .automation import RuleEngine, evaluate_video_detections, load_automation_text
from .detector import UltralyticsYOLOEDetector, parse_class_prompt
from .llm import (
    DEFAULT_ANTHROPIC_MODEL,
    DEFAULT_OPENAI_MODEL,
    DEFAULT_REPLICATE_MODEL,
    compile_automation_with_anthropic,
    compile_automation_with_openai,
    compile_automation_with_replicate,
)
from .video import process_video

__all__ = [
    "RuleEngine",
    "UltralyticsYOLOEDetector",
    "DEFAULT_ANTHROPIC_MODEL",
    "DEFAULT_OPENAI_MODEL",
    "DEFAULT_REPLICATE_MODEL",
    "compile_automation_with_anthropic",
    "compile_automation_with_openai",
    "compile_automation_with_replicate",
    "evaluate_video_detections",
    "load_automation_text",
    "parse_class_prompt",
    "process_video",
]

	"""Open-vocabulary video automations for Tiny Trigger."""

	from .automation import RuleEngine, evaluate_video_detections, load_automation_text
	from .detector import UltralyticsYOLOEDetector, parse_class_prompt
	from .llm import (
	DEFAULT_ANTHROPIC_MODEL,
	DEFAULT_OPENAI_MODEL,
	DEFAULT_REPLICATE_MODEL,
	compile_automation_with_anthropic,
	compile_automation_with_openai,
	compile_automation_with_replicate,
	)
	from .video import process_video

	__all__ = [
	"RuleEngine",
	"UltralyticsYOLOEDetector",
	"DEFAULT_ANTHROPIC_MODEL",
	"DEFAULT_OPENAI_MODEL",
	"DEFAULT_REPLICATE_MODEL",
	"compile_automation_with_anthropic",
	"compile_automation_with_openai",
	"compile_automation_with_replicate",
	"evaluate_video_detections",
	"load_automation_text",
	"parse_class_prompt",
	"process_video",
	]