From ff797c0741ce2ad6070ea2f81c38e3305cd3342c Mon Sep 17 00:00:00 2001
From: pharmapsychotic <96542870+pharmapsychotic@users.noreply.github.com>
Date: Mon, 20 Nov 2023 11:25:19 -0600
Subject: [PATCH] Fix Gradio issue and update engines for animation. (#262)

---
 setup.py                          | 4 ++--
 src/stability_sdk/animation.py    | 7 +++----
 src/stability_sdk/animation_ui.py | 2 +-
 3 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/setup.py b/setup.py
index e2a95f86..5f244633 100644
--- a/setup.py
+++ b/setup.py
@@ -10,7 +10,7 @@
 
 setup(
     name='stability-sdk',
-    version='0.8.4',
+    version='0.8.5',
     author='Stability AI',
     author_email='support@stability.ai',
     maintainer='Stability AI',
@@ -40,7 +40,7 @@
         ],
         'anim_ui': [
             'keyframed',
-            'gradio',
+            'gradio==3.50.2',
             'numpy',
             'opencv-python-headless',
             'tqdm',
diff --git a/src/stability_sdk/animation.py b/src/stability_sdk/animation.py
index d452cf97..00205a6d 100644
--- a/src/stability_sdk/animation.py
+++ b/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.")
diff --git a/src/stability_sdk/animation_ui.py b/src/stability_sdk/animation_ui.py
index d307eb4f..e24f3200 100644
--- a/src/stability_sdk/animation_ui.py
+++ b/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