Update app.py

app.py

@@ -1,7 +1,6 @@
 from transformers import VideoMAEImageProcessor, VideoMAEForVideoClassification, pipeline
 import gradio as gr
 import magic
-from moviepy.editor import VideoFileClip, TextClip, CompositeVideoClip
 
 pipe = pipeline("video-classification", model="anirudhmu/videomae-base-finetuned-soccer-action-recognition")
 
@@ -17,18 +16,7 @@ def predict_video(file_path):
     # Run inference
     results = pipe(file_path)
     label_to_score = {result["label"]: result["score"] for result in results}
-    
-    # Overlay prediction onto the video using moviepy
-    prediction_text = ', '.join([f"{k}: {v:.2f}" for k, v in label_to_score.items()])
-    clip = VideoFileClip(file_path)
-    txt_clip = TextClip(prediction_text, fontsize=24, color='white').set_pos('bottom').set_duration(clip.duration)
-    video = CompositeVideoClip([clip, txt_clip])
-    
-    # Saving video to temporary path and returning it
-    output_path = "/tmp/predicted_video.mp4"
-    video.write_videofile(output_path, codec='libx264')
-    
-    return output_path
+    return str(label_to_score)
 
-iface = gr.Interface(fn=predict_video, inputs="video", outputs="video")
-iface.launch()
+iface = gr.Interface(fn=predict_video, inputs="video", outputs="text")
+iface.launch()