Create app.py

app.py

@@ -0,0 +1,69 @@
+import os
+import pathlib
+import random
+import shlex
+import subprocess
+import gradio as gr
+from gradio import inputs, outputs
+import torch
+from huggingface_hub import snapshot_download
+from modelscope.pipelines import pipeline
+from modelscope.outputs import OutputKeys
+import boto3
+from botocore.client import Config
+import gradio as gr
+
+# Downloading and setting up the model
+model_dir = pathlib.Path('weights')
+if not model_dir.exists():
+    model_dir.mkdir()
+    snapshot_download('damo-vilab/modelscope-damo-text-to-video-synthesis',
+                      repo_type='model',
+                      local_dir=model_dir)
+
+s3_access_key = "juj22qxqxql7u2pl6nomgxu3ip7a"
+s3_secret_key = "j3uwidtozhboy5vczhymhzkkjsaumznnqlzck5zjs5qxgsung4ukk"
+s3_endpoint = "https://gateway.storjshare.io"
+
+s3 = boto3.client("s3", aws_access_key_id=s3_access_key, aws_secret_access_key=s3_secret_key, endpoint_url=s3_endpoint, config=Config(signature_version="s3v4"))
+
+# Function to generate video and upload it to Storj
+def generate_video(prompt: str, seed: int) -> str:
+    if seed == -1:
+        seed = random.randint(0, 1000000)
+    torch.manual_seed(seed)
+    result = pipe({'text': prompt})[OutputKeys.OUTPUT_VIDEO]
+    
+    # Upload video to Storj
+    bucket_name = "huggingface-demo"
+    # Add the code to upload the video to the Storj bucket
+    # and return the URL of the uploaded video
+
+    return result
+
+# Gradio Interface
+examples = [
+    ['An astronaut riding a horse.', 0],
+    ['A panda eating bamboo on a rock.', 0],
+    ['Spiderman is surfing.', 0],
+]
+
+# Import Storj Theme from the hub
+storj_theme = gr.Theme.from_hub("bethecloud/storj_theme")
+
+inputs = [
+    gr.inputs.Textbox(lines=1, placeholder="Enter your text here..."),
+    gr.inputs.Slider(minimum=-1, maximum=1000000, step=1, default=-1, label="Seed")
+]
+
+iface = gr.Interface(
+    fn=generate_video,
+    inputs=inputs,
+    outputs=gr.outputs.Video(),
+    theme=storj_theme,
+    allow_flagging=False,
+    examples=examples
+)
+
+## Run app
+iface.launch(share=True, debug=True)