|
|
|
|
|
import psutil |
|
|
|
import streamlit as st |
|
from audio2text.a2t import A2T |
|
from llm.llm import LLM_chain |
|
from streamlit_mic_recorder import mic_recorder |
|
|
|
llmchain = LLM_chain() |
|
|
|
def main(): |
|
mic = mic_recorder(start_prompt="Record",stop_prompt="Stop", just_once=True) |
|
|
|
if mic is not None: |
|
a2t = A2T(mic["bytes"]) |
|
text = a2t.predict() |
|
st.write(text) |
|
response = llmchain(entity=text, id=0) |
|
st.write(response if response is not None else "Nothing") |
|
del response |
|
|
|
|
|
if __name__ == "__main__": |
|
print('RAM memory % used:', psutil.virtual_memory()[2]) |
|
print('RAM Used (GB):', psutil.virtual_memory()[3]/1000000000) |
|
main() |
