CineAI commited on
Commit
d834bcd
·
verified ·
1 Parent(s): 1fe2644

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +21 -4
app.py CHANGED
@@ -2,6 +2,7 @@
2
 
3
  import platform
4
  import psutil
 
5
  # components from other authors
6
  from streamlit_mic_recorder import mic_recorder
7
 
@@ -13,13 +14,20 @@ from llm.llm_factory import LLM_Factory
13
 
14
  llm_model = LLM_Factory()
15
 
16
- trigger = {"lc": "small"}
17
-
18
  print(platform.node())
19
 
 
 
20
  def prepare_cor(input_text: str):
21
  return build_chain.build_command_chain().handle_command(input_text)
22
 
 
 
 
 
 
 
 
23
 
24
  def main():
25
  mic = mic_recorder(start_prompt="Record", stop_prompt="Stop", just_once=True)
@@ -28,14 +36,23 @@ def main():
28
  a2t = A2T(mic["bytes"])
29
  text = a2t.predict()
30
  print(text)
 
 
31
  # prepare_cor(input_text=text)
 
 
32
  llm = llm_model.create_llm(prompt_entity=text, prompt_id=1, trigger=trigger)
33
- response = llm.execution() if llm is not None else "Oops occurred some error. Please try again"
34
 
 
35
  t2a = T2A(response)
36
  t2a.autoplay()
37
 
38
 
39
  if __name__ == "__main__":
40
- print('RAM memory % used:', psutil.virtual_memory()[2]) # ~ 94 GB full memory
 
 
 
 
41
  main()
 
2
 
3
  import platform
4
  import psutil
5
+
6
  # components from other authors
7
  from streamlit_mic_recorder import mic_recorder
8
 
 
14
 
15
  llm_model = LLM_Factory()
16
 
 
 
17
  print(platform.node())
18
 
19
+ mac_address = platform.node()
20
+
21
  def prepare_cor(input_text: str):
22
  return build_chain.build_command_chain().handle_command(input_text)
23
 
24
+ # Якщо MAC адреса співпадає з адресою цього Space, тоді використовується hf endpoint, це базова перевірка для trigger.
25
+ # Від можливості використовувати AI agent також буде залежати trigger, якщо використовується, то ліпше використовувати lc ніж hf.
26
+ if mac_address[-5:] == os.environ.get('MAC_address'):
27
+ trigger = {"hf": "effective"}
28
+ else:
29
+ trigger = {"lc": "small"}
30
+
31
 
32
  def main():
33
  mic = mic_recorder(start_prompt="Record", stop_prompt="Stop", just_once=True)
 
36
  a2t = A2T(mic["bytes"])
37
  text = a2t.predict()
38
  print(text)
39
+
40
+ # Придумати як реалізувати команди
41
  # prepare_cor(input_text=text)
42
+
43
+ # Треба для lc реалізувати буде виклик очишення
44
  llm = llm_model.create_llm(prompt_entity=text, prompt_id=1, trigger=trigger)
45
+ response = llm.execution() if llm is not None else "Oops occurred some error. Please try again. Who is Jhon Galt!"
46
 
47
+ # Треба буде переписати клас, передавати текст не через __init__ а в autoplay.
48
  t2a = T2A(response)
49
  t2a.autoplay()
50
 
51
 
52
  if __name__ == "__main__":
53
+ print(f"Total Memory: {psutil.virtual_memory().total / (1024**3):.2f} GB")
54
+ print(f"Available Memory: {psutil.virtual_memory().available / (1024**3):.2f} GB")
55
+ print(f"CPU Cores: {psutil.cpu_count()}")
56
+ print(f"CPU Usage: {psutil.cpu_percent()}%")
57
+
58
  main()