diff --git a/paddlespeech/cli/base_commands.py b/paddlespeech/cli/base_commands.py index f9e2a55f..7210091a 100644 --- a/paddlespeech/cli/base_commands.py +++ b/paddlespeech/cli/base_commands.py @@ -125,9 +125,11 @@ class StatsCommand: "Here is the list of {} pretrained models released by PaddleSpeech that can be used by command line and python API" .format(self.task.upper())) self.show_support_models(pretrained_models) + return True except BaseException: print("Failed to get the list of {} pretrained models.".format( self.task.upper())) + return False # Dynamic import when running specific command diff --git a/paddlespeech/server/bin/paddlespeech_server.py b/paddlespeech/server/bin/paddlespeech_server.py index 175e8ffb..10a91d9b 100644 --- a/paddlespeech/server/bin/paddlespeech_server.py +++ b/paddlespeech/server/bin/paddlespeech_server.py @@ -18,7 +18,6 @@ from typing import List import uvicorn from fastapi import FastAPI -from starlette.middleware.cors import CORSMiddleware from prettytable import PrettyTable from starlette.middleware.cors import CORSMiddleware @@ -46,6 +45,7 @@ app.add_middleware( allow_methods=["*"], allow_headers=["*"]) + @cli_server_register( name='paddlespeech_server.start', description='Start the service') class ServerExecutor(BaseExecutor): @@ -177,7 +177,7 @@ class ServerStatsExecutor(): logger.info( "Here is the table of {} static pretrained models supported in the service.". format(self.task.upper())) - self.show_support_models(pretrained_models) + self.show_support_models(static_pretrained_models) return True