diff --git a/rtcllm/a2a.py b/rtcllm/a2a.py index c93217c..1f6c47a 100644 --- a/rtcllm/a2a.py +++ b/rtcllm/a2a.py @@ -2,6 +2,8 @@ import io from mini_omni.inference import OmniInference from av import AudioFrame +from aiortc import VideoStreamTrack, AudioStreamTrack + class LLMAudioStreamTrack(AudioStreamTrack): def __init__(self, ckpt_dir): super().__init__() diff --git a/rtcllm/rtc.py b/rtcllm/rtc.py index 604d6cf..d83c7a7 100644 --- a/rtcllm/rtc.py +++ b/rtcllm/rtc.py @@ -1,3 +1,6 @@ +import sys +sys.path.append('./mini_omni') + import asyncio import random import json