File size: 1,044 Bytes
cab263c
 
1581bbf
3c2d633
cab263c
4affef3
8685a84
 
f8dfb0f
d63135e
2980190
8685a84
 
cd60f99
8685a84
42bce02
2980190
 
42b403c
cd60f99
 
 
 
 
2980190
8685a84
 
 
 
05fa486
 
8685a84
3c2d633
05fa486
cd60f99
e991b7e
 
 
4affef3
 
3c2d633
4affef3
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
# version - ArticMonkey:19.03.24:1743

import psutil
import sys

import streamlit as st 

# my modules 
from audio_processing.A2T import A2T
from audio_processing.T2A import T2A
from llm.llm import LLM_chain

# libraries from other authors 
from streamlit_mic_recorder import mic_recorder
from streamlit_TTS import auto_play

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()
        response = llmchain(entity=text, id=0)
    
        audio, _, _ = T2A(response).get_audio()
        auto_play(audio)

        print(sys.getsizeof(response))
        print(sys.getsizeof(audio))
        
        del response
        del audio
        
    print('RAM memory % used after:', psutil.virtual_memory()[2])  
    print("Just checking : ", psutil.virtual_memory())
             

if __name__ == "__main__":
    print('RAM memory % used:', psutil.virtual_memory()[2]) # ~ 94 GB full memory
    main()