Update main_process/moe_router.py
Browse files
main_process/moe_router.py
CHANGED
|
@@ -23,6 +23,9 @@ from svision_client import (
|
|
| 23 |
extract_descripcion_escena
|
| 24 |
)
|
| 25 |
|
|
|
|
|
|
|
|
|
|
| 26 |
from asr_client import (
|
| 27 |
extract_audio_from_video,
|
| 28 |
diarize_audio,
|
|
@@ -66,6 +69,7 @@ class SalamandraClient:
|
|
| 66 |
)
|
| 67 |
|
| 68 |
def chat(self, prompt) -> str:
|
|
|
|
| 69 |
encodings = self.tokenizer(
|
| 70 |
prompt,
|
| 71 |
return_tensors="pt",
|
|
|
|
| 23 |
extract_descripcion_escena
|
| 24 |
)
|
| 25 |
|
| 26 |
+
from stools_client import (
|
| 27 |
+
get_from_prompt
|
| 28 |
+
)
|
| 29 |
from asr_client import (
|
| 30 |
extract_audio_from_video,
|
| 31 |
diarize_audio,
|
|
|
|
| 69 |
)
|
| 70 |
|
| 71 |
def chat(self, prompt) -> str:
|
| 72 |
+
return get_from_prompt(prompt)
|
| 73 |
encodings = self.tokenizer(
|
| 74 |
prompt,
|
| 75 |
return_tensors="pt",
|