Fix Gradio issue and update engines for animation. (#262)

setup.py

@@ -10,7 +10,7 @@
 
 setup(
     name='stability-sdk',
-    version='0.8.4',
+    version='0.8.5',
     author='Stability AI',
     author_email='[email protected]',
     maintainer='Stability AI',
@@ -40,7 +40,7 @@
         ],
         'anim_ui': [
             'keyframed',
-            'gradio',
+            'gradio==3.50.2',
             'numpy',
             'opencv-python-headless',
             'tqdm',

src/stability_sdk/animation.py

@@ -35,7 +35,7 @@
 logger = logging.getLogger(__name__)
 logger.setLevel(level=logging.INFO)
 
-DEFAULT_MODEL = 'stable-diffusion-v1-5'
+DEFAULT_MODEL = 'stable-diffusion-xl-1024-v0-9'
 TRANSLATION_SCALE = 1.0/200.0 # matches Disco and Deforum
 
 docstring_bordermode = ( 
@@ -61,9 +61,8 @@ class BasicSettings(param.Parameterized):
         default=DEFAULT_MODEL, 
         check_on_set=False, # allow old and new models without raising ValueError
         objects=[
-            "stable-diffusion-v1-5", "stable-diffusion-512-v2-1", "stable-diffusion-768-v2-1", 
-            "stable-diffusion-depth-v2-0", "stable-diffusion-xl-beta-v2-2-2", "stable-diffusion-xl-1024-v1-0",
-            "custom"
+            "stable-diffusion-512-v2-1", "stable-diffusion-xl-beta-v2-2-2", "stable-diffusion-xl-1024-v0-9", 
+            "stable-diffusion-xl-1024-v1-0", "custom"
         ]
     )
     custom_model = param.String(default="", doc="Identifier of custom model to use.")

src/stability_sdk/animation_ui.py

@@ -158,7 +158,7 @@ def list_projects(cls) -> List["Project"]:
 negative_prompt_weight = -1.0
 
 controls: Dict[str, gr.components.Component] = {}
-header = gr.HTML("", show_progress=False)
+header = gr.HTML("")
 interrupt = False
 last_interp_factor = None
 last_interp_mode = None