Add simple progress reporting.

app.py

@@ -89,9 +89,14 @@ def on_model_change(model):
         model_type = None
 
 
-def convert_model(preprocessor, model, model_coreml_config, compute_units, precision, tolerance, output, use_past=False, seq2seq=None):
+def convert_model(preprocessor, model, model_coreml_config,
+                  compute_units, precision, tolerance, output,
+                  use_past=False, seq2seq=None,
+                  progress=None, progress_start=0.1, progress_end=0.8):
     coreml_config = model_coreml_config(model.config, use_past=use_past, seq2seq=seq2seq)
 
+    model_label = "model" if seq2seq is None else seq2seq
+    progress(progress_start, desc=f"Converting {model_label}")
     mlmodel = export(
         preprocessor,
         model,
@@ -109,12 +114,14 @@ def convert_model(preprocessor, model, model_coreml_config, compute_units, preci
 
     mlmodel.save(filename)
 
+    progress(progress_end * 0.8, desc=f"Validating {model_label}")
     if tolerance is None:
         tolerance = coreml_config.atol_for_validation
     validate_model_outputs(coreml_config, preprocessor, model, mlmodel, tolerance)
+    progress(progress_end, desc=f"Done converting {model_label}")
 
 
-def convert(model, task, compute_units, precision, tolerance, framework):
+def convert(model, task, compute_units, precision, tolerance, framework, progress=gr.Progress()):
     model = url_to_model_id(model)
     compute_units = compute_units_mapping[compute_units]
     precision = precision_mapping[precision]
@@ -127,12 +134,13 @@ def convert(model, task, compute_units, precision, tolerance, framework):
     output = output/f"{precision}_model.mlpackage"
 
     try:
+        progress(0, desc="Downloading model")
+
         preprocessor = get_preprocessor(model)
         model = FeaturesManager.get_model_from_feature(task, model, framework=framework)
         _, model_coreml_config = FeaturesManager.check_supported_model_or_raise(model, feature=task)
 
         if task in ["seq2seq-lm", "speech-seq2seq"]:
-            # Convert encoder / decoder
             convert_model(
                 preprocessor,
                 model,
@@ -141,8 +149,12 @@ def convert(model, task, compute_units, precision, tolerance, framework):
                 precision,
                 tolerance,
                 output,
-                seq2seq="encoder"
+                seq2seq="encoder",
+                progress=progress,
+                progress_start=0.1,
+                progress_end=0.45,
             )
+            progress(0.6, desc="Converting decoder")
             convert_model(
                 preprocessor,
                 model,
@@ -151,7 +163,10 @@ def convert(model, task, compute_units, precision, tolerance, framework):
                 precision,
                 tolerance,
                 output,
-                seq2seq="decoder"
+                seq2seq="decoder",
+                progress=progress,
+                progress_start=0.45,
+                progress_end=0.8,
             )
         else:
             convert_model(
@@ -162,9 +177,12 @@ def convert(model, task, compute_units, precision, tolerance, framework):
                 precision,
                 tolerance,
                 output,
+                progress=progress,
+                progress_end=0.8,
             )
 
         # TODO: push to hub, whatever
+        progress(1, "Done")
         return "Done"
     except Exception as e:
         return error_str(e)