Update app.py

app.py

@@ -2,6 +2,18 @@ import cv2
 import streamlit as st
 from fastai.vision.all import *
 import base64
+from streamlit_webrtc import webrtc_streamer
+import av
+import time
+
+
+def video_frame_callback(frame):
+    time.sleep(1)
+    return frame
+
+
+webrtc_streamer(key="example", video_frame_callback=video_frame_callback)
+
 st.markdown('<h1 style="color:black;">BDI Image classification model</h1>', unsafe_allow_html=True)
 learn = load_learner('export (5).pkl')
 upload = st.file_uploader('Insert image for classification', type=['png','jpg'])