Skip to content

Commit c1f8761

Browse files
authored
fix app_ui (#1036)
1 parent ca8c468 commit c1f8761

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

swift/llm/app_ui.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -44,9 +44,9 @@ def model_generation(query: str) -> Iterator[str]:
4444
send.click(model_generation, inputs=[input_box], outputs=[output_box])
4545
# Compatible with InferArguments
4646
share = getattr(args, 'share', False)
47-
server_name = getattr(args, 'server_name', '127.0.0.1')
48-
server_port = getattr(args, 'server_port', 7860)
49-
demo.queue().launch(height=1000, share=share, server_name=server_name, server_port=server_port)
47+
host = getattr(args, 'host', '127.0.0.1')
48+
port = getattr(args, 'port', 7860)
49+
demo.queue().launch(height=1000, share=share, server_name=host, server_port=port)
5050

5151

5252
def gradio_chat_demo(args: AppUIArguments) -> None:
@@ -84,9 +84,9 @@ def model_chat(query: str, history: History) -> Iterator[Tuple[str, History]]:
8484
clear_history.click(fn=clear_session, inputs=[], outputs=[chatbot], queue=False)
8585
# Compatible with InferArguments
8686
share = getattr(args, 'share', False)
87-
server_name = getattr(args, 'server_name', '127.0.0.1')
88-
server_port = getattr(args, 'server_port', 7860)
89-
demo.queue().launch(height=1000, share=share, server_name=server_name, server_port=server_port)
87+
host = getattr(args, 'host', '127.0.0.1')
88+
port = getattr(args, 'port', 7860)
89+
demo.queue().launch(height=1000, share=share, server_name=host, server_port=port)
9090

9191

9292
def llm_app_ui(args: AppUIArguments) -> None:

0 commit comments

Comments
 (0)