video / app.py
seawolf2357's picture
Update app.py
1cd16bc verified
raw
history blame
1.36 kB
import os
import gradio as gr
from huggingface_hub import HfApi
# ν™˜κ²½ λ³€μˆ˜μ—μ„œ Hugging Face 토큰 읽기
hf_token = os.getenv("HF_TOKEN")
# Hugging Face API μ΄ˆκΈ°ν™”
api = HfApi()
def upload_file_to_hf_space(uploaded_file):
# μ‚¬μš©μž ID와 슀페이슀 이름 μ„€μ •
user_id = "seawolf2357"
space_name = "video" # 슀페이슀 이름을 여기에 μž…λ ₯ν•˜μ„Έμš”.
repo_id = f"{user_id}/{space_name}"
# 파일λͺ… μ„€μ • 및 μ €μž₯
file_path = uploaded_file.name
with open(file_path, "wb") as f:
f.write(uploaded_file.read())
# 파일 μœ ν˜• 검사 (MP4 νŒŒμΌμΈμ§€ 확인)
if not file_path.endswith('.mp4'):
return "Please upload an MP4 file."
# Hugging Face Spaces에 파일 μ—…λ‘œλ“œ
response = api.upload_file(
path_or_fileobj=file_path,
path_in_repo=os.path.basename(file_path),
repo_id=repo_id,
token=hf_token,
)
# μ—…λ‘œλ“œλœ 파일의 URL λ°˜ν™˜
uploaded_file_url = f"https://huggingface.co/spaces/{repo_id}/blob/main/{os.path.basename(file_path)}"
return uploaded_file_url
# Gradio μΈν„°νŽ˜μ΄μŠ€ μ„€μ •
iface = gr.Interface(
fn=upload_file_to_hf_space,
inputs=gr.File(label="Upload your MP4 file"),
outputs="text",
title="MP4 File Upload to Hugging Face Spaces",
description="Upload an MP4 file and