Update app.py

app.py

@@ -74,33 +74,27 @@ class ShopliftingPrediction:
         
         cv2.rectangle(frame, (0, 0), (640, 40), (255, 255, 255), -1)
         cv2.putText(frame, self.message, (1, 30), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 0, 0), 1, cv2.LINE_AA)
-        return frame, frames_queue, self.message
+        return frame
 
 def inference(model_path):
     shoplifting_prediction = ShopliftingPrediction(model_path, 64, 64, 30)
     shoplifting_prediction.load_model()
     
     def process_video(video_frame):
-        frames_queue = []
-        previous = video_frame
-        frame, frames_queue, message = shoplifting_prediction.process_frame(video_frame, frames_queue, previous)
-        return frame, message
+        frame = shoplifting_prediction.process_frame(video_frame, [], video_frame)
+        return frame
 
     return process_video
 
 model_path = 'lrcn_160S_90_90Q.h5'
 process_video = inference(model_path)
 
-def process_output(video):
-    for frame in video:
-        frame, message = process_video(frame)
-        yield frame, message
-
 iface = gr.Interface(
-    fn=process_output,
-    inputs=gr.Video(),
-    outputs=["video", "text"],
-    live=True
+    fn=process_video,
+    inputs=gr.inputs.Video(type="numpy"),
+    outputs="video",
+    live=True,
+    capture_session=True  # Ensures the video stream is captured properly
 )
 
 iface.launch()