besarismaili's picture
1
ff3c6aa
raw
history blame
926 Bytes
#@title Connect to the Stability API
import os
from stability_sdk.api import Context
from stability_sdk.animation_ui import create_ui
# @markdown To get your API key visit https://dreamstudio.ai/account
STABILITY_HOST = "grpc.stability.ai:443" #@param {type:"string"}
STABILITY_KEY = os.getenv("STABILITY_KEY")
# Connect to Stability API
context = Context(STABILITY_HOST, STABILITY_KEY)
# Test the connection
context.get_user_info()
print("Connection successful!")
#@title Animation UI
show_ui_in_notebook = True #@param {type:"boolean"}
outputs_path = r'C:\Work\tst\SAnim'
# Create UI
interface = create_ui(context, outputs_path)
# Set concurrency_limit on the Interface
interface = gr.Interface(fn=interface.fn, inputs=interface.inputs, outputs=interface.outputs, concurrency_limit=2)
# Launch the UI
interface.launch(show_api=False, debug=True, inline=show_ui_in_notebook, height=768, share=True, show_error=True)