Update

app.py

@@ -59,7 +59,8 @@ ai_prefix = "### Assistant"
 stop_list = [f"{human_prefix}:"]
 
 if "beluga" in model_loc.lower():
-    prompt_template = dedent("""
+    prompt_template = dedent(
+        """
     ### System:
     You are Stable Beluga, an AI that follows instructions extremely well. Help as much as you can.
     Let's think step by step.
@@ -67,7 +68,8 @@ if "beluga" in model_loc.lower():
     ### User: {question}
 
     ### Assistant:
-    """).lstrip()
+    """
+    ).lstrip()
     human_prefix = "### User"
     ai_prefix = "### Assistant"
     stop_list = [f"{human_prefix}:"]
@@ -380,6 +382,8 @@ else:
 server_port = 7860
 if "forindo" in platform.node():
     server_port = 7861
-block.queue(max_size=5).launch(debug=True, server_name="0.0.0.0", server_port=server_port)
+block.queue(max_size=5).launch(
+    debug=True, server_name="0.0.0.0", server_port=server_port
+)
 
 # block.queue(max_size=5).launch(debug=True, server_name="0.0.0.0")

dl_model.py

@@ -20,7 +20,9 @@ app = typer.Typer(
 
 def _version_callback(value: bool) -> None:
     if value:
-        typer.echo(f"{app.info.name} v.{__version__} -- download models for given url(s)")
+        typer.echo(
+            f"{app.info.name} v.{__version__} -- download models for given url(s)"
+        )
         raise typer.Exit()
 
 
@@ -28,8 +30,8 @@ def _version_callback(value: bool) -> None:
 def main(
     urls: str = typer.Argument(  # pylint: disable=unused-argument
         "",
-       help=f"one or more urls (default {url})",
-       show_default=False,
+        help=f"one or more urls (default {url})",
+        show_default=False,
     ),
     version: Optional[bool] = typer.Option(  # pylint: disable=unused-argument
         None,
@@ -46,15 +48,7 @@ def main(
         help="dir to save downloaded models (default models)",
     ),
 ):
-    """
-    Define main.
-
-    Args:
-    ----
-    version: verson and help info.
-    urls: model urls separated by space(s) (default url.
-    model_dir: location to save downlaoded models (default models).
-    """
+    """Download a model or model given url(s)."""
     logger.trace(f"{urls}")
     if model_dir is None:
         model_dir = "models"