Spaces:
Runtime error
Runtime error
File size: 2,201 Bytes
ff3c6aa d5d86d7 998b8a0 ff3c6aa 0b11a54 1c71d5b 998b8a0 0b11a54 ddabfc0 ff3c6aa 5273c29 ddabfc0 5273c29 0b11a54 5273c29 0b11a54 5273c29 0b11a54 4230814 0937600 4230814 02d0fb2 4230814 02d0fb2 abfd2c7 c02f1e3 5273c29 1f14d54 0b11a54 4230814 c46c1c7 4230814 9365fbc 4230814 5273c29 fe2bf46 c02f1e3 |
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 74 75 76 77 78 |
import os
import io
import base64
import gradio as gr
from stability_sdk.api import Context
from stability_sdk.animation import AnimationArgs, Animator
from dotenv import load_dotenv
load_dotenv(".env")
STABILITY_HOST = "grpc.stability.ai:443"
#STABILITY_KEY = os.getenv("STABILITY_KEY")
def anim(f_promt, s_promt, stability_key):
# Connect to Stability API
context = Context(STABILITY_HOST, stability_key)
# Test the connection
context.get_user_info()
print("Connection successfuly!")
# Configure the animation
args = AnimationArgs()
args.interpolate_prompts = True
args.locked_seed = True
args.max_frames = 20
args.seed = 42
args.strength_curve = "0:(0)"
args.diffusion_cadence_curve = "0:(4)"
args.cadence_interp = "film"
animation_prompts = {
0: f_promt,
10: s_promt,
}
negative_prompt = ""
# Create Animator object to orchestrate the rendering
animator = Animator(
api_context=context,
animation_prompts=animation_prompts,
negative_prompt=negative_prompt,
args=args
)
# Render each frame of animation
for idx, frame in enumerate(animator.render()):
file_path = os.path.join("output", f"frame_{idx:05d}.png")
frame.save(file_path)
yield file_path
def file():
# Get current working directory
app_dir = os.getcwd()
# Set output folder for files
output_dir = os.path.join(app_dir, "moutput")
if not os.path.exists(output_dir):
os.makedirs(output_dir)
print("Created folder at:"+output_dir)
with gr.Blocks() as demo:
gr.Markdown("Stability Animation")
f_promt = gr.Textbox(label="First Prompt", value="a photo of a cute cat")
s_promt = gr.Textbox(label="Second Prompt", value="a photo of a cute dog")
stability_key = gr.Textbox(label="Stability Key", value="")
outimg = gr.Files(label="Generated Files")
btn = gr.Button('Anim')
fbtn = gr.Button('Get Dir')
files = gr.Text()
fbtn.click(fn=file, inputs=[], outputs=[files])
btn.click(fn=anim, inputs=[f_promt, s_promt, stability_key], outputs=[outimg],api_name="AnimAPI")
demo.launch() |