Update app.py

app.py

@@ -103,6 +103,12 @@ def server(input, output, session):
     def translate():
         global t
         model_path = Path("/tmp") / input.model()
+
+        if not model_path.exists():
+            model_download_output(
+                model_name="quickmt/" + input.model(),
+                output_dir=Path("/tmp") / input.model(),
+            )
         try:
             if t is None or str(input.model()) != str(Path(t.model_path).name):
                 print(f"Loading model {input.model()}")
@@ -124,14 +130,10 @@ def server(input, output, session):
         except:
             return [
                 ui.value_box(
-                    title=f"Ensure model is downloaded",
+                    title=f"Unexpected error",
                     value="Failed to load model",
                     showcase=icon("bug"),
                 ),
-                ui.input_action_button(
-                    "quickmt_model_download", "Download Model", class_="btn-primary"
-                ),
-                ui.output_ui("model_download_output"),
             ]
 
 app = App(app_ui, server)