Update app.py

app.py

@@ -1,4 +1,3 @@
-# app.py
 import cv2
 import numpy as np
 import torch
@@ -12,7 +11,7 @@ feature_extractor = AutoFeatureExtractor.from_pretrained(model_name)
 
 def process_frame(frame):
     # Convert frame to tensor
-    inputs = feature_extractor(images=frame, return_tensors="pt")
+    inputs = feature_extractor(images=[frame], return_tensors="pt")
     outputs = model(**inputs)
     predictions = torch.nn.functional.softmax(outputs.logits, dim=-1)
     return predictions
@@ -29,17 +28,11 @@ def main():
             st.write("Failed to capture video")
             break
         else:
-            predictions = process_frame(frame)
             frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
+            predictions = process_frame(frame)
             frame_placeholder.image(frame, channels="RGB")
     cap.release()
     
 if __name__ == "__main__":
     main()
 
-# requirements.txt
-opencv-python
-numpy
-torch
-transformers
-streamlit