WlSL / test.py
Ldrago116's picture
Upload 4 files
b8f92c8 verified
raw
history blame contribute delete
818 Bytes
import gradio as gr
import os
from utils import classfication
source_video = "videos/main.mp4"
def video_identity(video):
print(type(video))
try:
os.makedirs(os.path.dirname(source_video ), exist_ok=True)
os.replace(video, source_video )
except Exception as e:
print(f"Error: {e}")
predection =classfication()
files = os.listdir(source_video)
# Iterate through each file and delete
for file in files:
file_path = os.path.join(source_video, file)
if os.path.isfile(file_path):
os.remove(file_path)
return predection
demo = gr.Interface(video_identity,
gr.Video(),
outputs="text"
)
if __name__ == "__main__":
demo.launch(share=True)