Upload 3 files

app.py

@@ -167,7 +167,7 @@ class GuiSD:
         self.last_load = datetime.now()
         self.inventory = []
 
-    def update_storage_models(self, storage_floor_gb=32, required_inventory_for_purge=3):
+    def update_storage_models(self, storage_floor_gb=24, required_inventory_for_purge=3):
         while get_used_storage_gb() > storage_floor_gb:
             if len(self.inventory) < required_inventory_for_purge:
                 break
@@ -780,7 +780,7 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', elem_id="main", fill_width=True, cs
                 with gr.Column(scale=1):
 
                     def update_task_options(model_name, task_name):
-                        new_choices = MODEL_TYPE_TASK[get_model_type(model_name)]
+                        new_choices = MODEL_TYPE_TASK[get_model_type(valid_model_name(model_name))]
 
                         if task_name not in new_choices:
                             task_name = "txt2img"
@@ -863,7 +863,6 @@ with gr.Blocks(theme='NoCrypt/miku@>=1.2.2', elem_id="main", fill_width=True, cs
                         object_fit="contain",
                         # height="auto",
                         interactive=False,
-                        #preview=False,
                         preview=False,
                         show_share_button=False,
                         show_download_button=True,

modutils.py

@@ -172,7 +172,7 @@ class ModelInformation:
         self.download_url = json_data.get("downloadUrl", "")
         self.model_url = f"https://civitai.com/models/{self.model_id}?modelVersionId={self.model_version_id}"
         self.filename_url = next(
-            (v.get("name", "") for v in json_data.get("files", []) if str(self.model_version_id) in v.get("downloadUrl", "")), ""
+            (v.get("name", "") for v in reversed(json_data.get("files", [])) if str(self.model_version_id) in v.get("downloadUrl", "")), ""
         )
         self.filename_url = self.filename_url if self.filename_url else ""
         self.description = json_data.get("description", "")

requirements.txt

@@ -13,4 +13,5 @@ translatepy
 timm
 rapidfuzz
 sentencepiece
-unidecode
+unidecode
+ultralytics==8.3.47