VeuReu commited on
Commit
fe4d381
·
verified ·
1 Parent(s): 9b265a5

Update main_process/moe_router.py

Browse files
Files changed (1) hide show
  1. main_process/moe_router.py +4 -0
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",