TAneKAnz's picture
update
82e48c9 verified
raw history blame
No virus
916 Bytes
import cv2
import gradio as gr
import os
def convert_to_grayscale(video_file):
cap = cv2.VideoCapture(video_file)
output_file = "output.mp4"
width = int(cap.get(cv2.CAP_PROP_FRAME_WIDTH))
height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
fps = int(cap.get(cv2.CAP_PROP_FPS))
fourcc = cv2.VideoWriter_fourcc(*'mp4v')
out = cv2.VideoWriter(output_file, fourcc, fps, (width, height), isColor=False)
while(cap.isOpened()):
ret, frame = cap.read()
if not ret:
break
gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)
out.write(gray)
cap.release()
out.release()
return output_file
demo = gr.Interface(
fn=convert_to_grayscale,
title="Video Upload and Display",
inputs=gr.Video(label="Upload Video", height=500, width=500),
outputs=gr.Video(label="Grayscale Video", height=500, width=500),
)
demo.launch()