YingxuHe commited on
Commit
a42d133
·
verified ·
1 Parent(s): ef2ea8a

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +17 -15
utils.py CHANGED
@@ -9,6 +9,22 @@ from sshtunnel import SSHTunnelForwarder
9
 
10
  local_port = int(os.getenv('LOCAL_PORT'))
11
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
12
  @st.cache_resource()
13
  def load_model():
14
  openai_api_key = os.getenv('API_KEY')
@@ -56,18 +72,4 @@ def bytes_to_array(audio_bytes):
56
  io.BytesIO(audio_bytes),
57
  sr=16000
58
  )
59
- return audio_array
60
-
61
- @st.cache_resource()
62
- def start_server():
63
- pkey = paramiko.RSAKey.from_private_key(io.StringIO(os.getenv('PRIVATE_KEY')))
64
-
65
- server = SSHTunnelForwarder(
66
- ssh_address_or_host=os.getenv('SERVER_DNS_NAME'),
67
- ssh_username="ec2-user",
68
- ssh_pkey=pkey,
69
- local_bind_address=("127.0.0.1", local_port),
70
- remote_bind_address=("127.0.0.1", 8000)
71
- )
72
- server.start()
73
- return server
 
9
 
10
  local_port = int(os.getenv('LOCAL_PORT'))
11
 
12
+
13
+ @st.cache_resource()
14
+ def start_server():
15
+ pkey = paramiko.RSAKey.from_private_key(io.StringIO(os.getenv('PRIVATE_KEY')))
16
+
17
+ server = SSHTunnelForwarder(
18
+ ssh_address_or_host=os.getenv('SERVER_DNS_NAME'),
19
+ ssh_username="ec2-user",
20
+ ssh_pkey=pkey,
21
+ local_bind_address=("127.0.0.1", local_port),
22
+ remote_bind_address=("127.0.0.1", 8000)
23
+ )
24
+ server.start()
25
+ return server
26
+
27
+
28
  @st.cache_resource()
29
  def load_model():
30
  openai_api_key = os.getenv('API_KEY')
 
72
  io.BytesIO(audio_bytes),
73
  sr=16000
74
  )
75
+ return audio_array