Update app.py

app.py

@@ -97,7 +97,7 @@ def predict_emotion(frame):
         frame = predict_image(frame)
     else:
         # Video frame input
-        frame = predict_video_frame(frame)
+        frame = predict_video_frame(frame.read())  # Convert NamedString to bytes
     return frame
 
 def predict_image(frame):
@@ -128,32 +128,10 @@ def predict_image(frame):
 def predict_video_frame(frame):
     return predict_image(frame)
 
-# demo = gr.Interface(
-#     fn=predict_emotion,
-#     inputs=gr.inputs.Video(type="numpy"),  # Accepts video file input
-#     outputs=gr.outputs.Video("output.mp4"),  # Outputs processed video
-#     examples=[
-#         os.path.join(os.path.dirname(__file__), "images/chandler.jpeg"),
-#         os.path.join(os.path.dirname(__file__), "images/janice.jpeg"),
-#         os.path.join(os.path.dirname(__file__), "images/joey.jpeg"),
-#         os.path.join(os.path.dirname(__file__), "images/phoebe.jpeg"),
-#         os.path.join(os.path.dirname(__file__), "images/rachel_monica.jpeg"),
-#         os.path.join(os.path.dirname(__file__), "images/ross.jpeg"),
-#         os.path.join(os.path.dirname(__file__), "images/gunther.jpeg"),
-#         os.path.join(os.path.dirname(__file__), "images/joey_shocked.mp4")     
-#     ],
-#     title="How are you feeling?",
-#     theme="shivi/calm_seafoam"
-# )
-
 demo = gr.Interface(
     fn = predict_emotion,
-    inputs = [gr.components.File()],
-    outputs = gr.components.Image(),
-    # gr.components.Image(label="Predicted Emotion"),
-    # gr.components.Label(num_top_classes=2, label="Top 2 Probabilities")
-    #flagging_options=["blurry", "incorrect", "other"],
-    
+    inputs = [gr.inputs.Image(type="numpy"), gr.inputs.Video()],  # Corrected input components
+    outputs = gr.outputs.Image(),
     examples = [
         [os.path.join(os.path.dirname(__file__), "images/chandler.jpeg")],
         [os.path.join(os.path.dirname(__file__), "images/janice.jpeg")],
@@ -162,12 +140,11 @@ demo = gr.Interface(
         [os.path.join(os.path.dirname(__file__), "images/rachel_monica.jpeg")],
         [os.path.join(os.path.dirname(__file__), "images/ross.jpeg")],
         [os.path.join(os.path.dirname(__file__), "images/gunther.jpeg")],
-        [os.path.join(os.path.dirname(__file__), "images/joey_shocked.mp4")]
+        [open(os.path.join(os.path.dirname(__file__), "images/joey_shocked.mp4"), "rb")]  # Open file in binary mode
     ],
     title = "How are you feeling?",
     theme = "shivi/calm_seafoam"
 )
 
-
 if __name__ == "__main__":
     demo.launch()