Spaces:
Runtime error
Runtime error
File size: 2,414 Bytes
aa6df5a 06945c0 aa6df5a 06945c0 aa6df5a e63a5af 3eba7f6 a1c24f4 aa6df5a 06945c0 aa6df5a 06945c0 aa6df5a 06945c0 aa6df5a 06945c0 aa6df5a 06945c0 aa6df5a 06945c0 aa6df5a 1b03567 06945c0 aa6df5a 06945c0 1b03567 aa6df5a e63a5af |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 |
import os
import gradio as gr
from modelscope.pipelines import pipeline
from modelscope.outputs import OutputKeys
import requests
import tempfile
import boto3
from botocore.exceptions import NoCredentialsError
# Read S3 keys from environment variables
s3_access_key = "juj22qxqxql7u2pl6nomgxu3ip7a"
s3_secret_key = "j3uwidtozhboy5vczhymhzkkjsaumznnqlzck5zjs5qxgsung4ukk"
s3_endpoint = "https://gateway.storjshare.io"
# Initialize Storj with boto3
s3 = boto3.client(
"s3",
endpoint_url=s3_endpoint,
aws_access_key_id=s3_access_key,
aws_secret_access_key=s3_secret_key,
)
p = pipeline("text-to-video-synthesis", "damo/text-to-video-synthesis")
async def generate_video(text_input: str, duration: int) -> (str, str):
test_text = {'text': text_input, 'duration': duration}
output_video_path = p(test_text,)[OutputKeys.OUTPUT_VIDEO]
# Upload to Storj
with open(output_video_path, "rb") as f:
video_key = os.path.basename(output_video_path)
s3.upload_fileobj(f, "huggingface-demo", video_key)
# Generate signed URL for uploaded video
try:
storj_video_url = s3.generate_presigned_url(
"get_object",
Params={"Bucket": "huggingface-demo", "Key": video_key},
ExpiresIn=3600,
)
except NoCredentialsError:
print("Credentials not available")
return None
# Download the video from Storj to a temporary file
response = requests.get(storj_video_url, stream=True)
response.raise_for_status()
with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as temp_file:
for chunk in response.iter_content(chunk_size=8192):
temp_file.write(chunk)
local_video_path = temp_file.name
return local_video_path, storj_video_url
# Use the custom Gradio theme
storj_theme = gr.Theme.from_hub("bethecloud/storj_theme")
iface = gr.Interface(
fn=generate_video,
inputs=[
gr.inputs.Textbox(lines=3, label="Input text"),
gr.inputs.Slider(minimum=1, maximum=60, step=1, default=5, label="Duration (seconds)")
],
outputs=[
gr.outputs.Video(label="Generated Video"),
gr.outputs.Textbox(label="Storj Video URL")
],
title="Text to Video to Storj",
theme=storj_theme # Pass the theme to the Interface
)
iface.launch(share=False, debug=True) # Set share to False when deploying to Hugging Face Spaces |