warhawkmonk commited on
Commit
da72d70
·
verified ·
1 Parent(s): 6813803

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import streamlit as st
2
- from streamlit_webrtc import webrtc_streamer, VideoTransformerBase
3
  import av
4
  import threading
5
  import numpy as np
@@ -32,7 +32,10 @@ def base64_to_image(base64_string):
32
  # Open image with PIL
33
  img = Image.open(image_buffer)
34
  return img
35
-
 
 
 
36
  # Initialize session state keys if not present
37
  if 'current_image' not in st.session_state:
38
  st.session_state['current_image'] = None
@@ -79,10 +82,11 @@ with col1:
79
  }
80
  ctx = webrtc_streamer(
81
  key="example",
 
82
  video_transformer_factory=lambda: VideoTransformer(st.session_state),
83
  media_stream_constraints={"video": True, "audio": False},
84
  async_transform=True,
85
- rtc_configuration=rtc_configuration,
86
 
87
  )
88
 
 
1
  import streamlit as st
2
+ from streamlit_webrtc import webrtc_streamer, VideoTransformerBase, WebRtcMode
3
  import av
4
  import threading
5
  import numpy as np
 
32
  # Open image with PIL
33
  img = Image.open(image_buffer)
34
  return img
35
+ def get_ice_servers():
36
+ # Using Google's public STUN server
37
+ return [{"urls": ["stun:stun.l.google.com:19302"]}]
38
+
39
  # Initialize session state keys if not present
40
  if 'current_image' not in st.session_state:
41
  st.session_state['current_image'] = None
 
82
  }
83
  ctx = webrtc_streamer(
84
  key="example",
85
+ mode=WebRtcMode.SENDRECV,
86
  video_transformer_factory=lambda: VideoTransformer(st.session_state),
87
  media_stream_constraints={"video": True, "audio": False},
88
  async_transform=True,
89
+ rtc_configuration={"iceServers": get_ice_servers()},
90
 
91
  )
92