quality

app.py

@@ -63,26 +63,26 @@ def run(
         )
         return
 
-    total_size = sum(
-        file.size
-        for file in model_info.siblings
-        if file.rfilename.endswith(".pt") or file.rfilename.endswith(".bin")
-    )
-    if total_size > MAX_REPO_SIZE:
-        yield _log(
-            f"""
-        ### Unprocessable 😢😢😢
-        
-        Model {model_id} is too big and cannot be processed in this Space. This Space needs to be able to load the model in memory before converting it. To avoid a memory issue, we do not process models bigger than {MAX_REPO_SIZE}b.
-
-        You have 2 options:
-        - [Duplicate this Space](https://huggingface.co/spaces/Wauplin/bloomz.cpp-converter?duplicate=true) and assign a bigger machine. You will need to set 'MAX_REPO_SIZE' as a secret to overwrite the default value. Once you are done, remove the upgraded hardware and/or delete the Space.
-        - Manually convert the weights by following [this guide](https://github.com/NouamaneTazi/bloomz.cpp#usage).
-        """
+    try:
+        total_size = sum(
+            file.size
+            for file in model_info.siblings
+            if file.rfilename.endswith(".pt") or file.rfilename.endswith(".bin")
         )
-        return
+        if total_size > MAX_REPO_SIZE:
+            yield _log(
+                f"""
+            ### Unprocessable 😢😢😢
+
+            Model {model_id} is too big and cannot be processed in this Space. This Space needs to be able to load the model in memory before converting it. To avoid a memory issue, we do not process models bigger than {MAX_REPO_SIZE}b.
+
+            You have 2 options:
+            - [Duplicate this Space](https://huggingface.co/spaces/Wauplin/bloomz.cpp-converter?duplicate=true) and assign a bigger machine. You will need to set 'MAX_REPO_SIZE' as a secret to overwrite the default value. Once you are done, remove the upgraded hardware and/or delete the Space.
+            - Manually convert the weights by following [this guide](https://github.com/NouamaneTazi/bloomz.cpp#usage).
+            """
+            )
+            return
 
-    try:
         with TemporaryDirectory() as cache_folder:
             convert_progress = Generator(
                 convert(
@@ -147,6 +147,7 @@ def run(
             yield _log(
                 f"""
             ### Success 🔥
+
             Yay! This model was successfully converted! Make sure to let the repo owner know about it and review your PR. You might need to complete the PR manually, especially to add information in the model card.
             """
             )