Update app.py
Browse files
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(
|
|
|
|
|
|
|
|
|
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()
|