fix

Dockerfile

@@ -17,4 +17,4 @@ COPY --chown=user . $HOME/app
 
 EXPOSE 7860
 
-CMD ["sh", "-c", "python3 server.py --notebook --mlock"]
+CMD ["sh", "-c", "python3 server.py --notebook --listen"]

requirements.txt

@@ -2,8 +2,7 @@ accelerate==0.19.0
 colorama
 datasets
 flexgen==0.1.7
-gradio_client==0.2.5
-gradio==3.31.0
+gradio==3.32.0
 markdown
 numpy
 pandas

server.py

@@ -786,13 +786,15 @@ def create_interface():
         # Extensions block
         extensions_module.create_extensions_block()
 
-    print("start to Launch the interface")
+    logging.info('start to Launch the interface')
     # Launch the interface
     shared.gradio['interface'].queue()
+    logging.info('start to queue')
     if shared.args.listen:
-        shared.gradio['interface'].launch(prevent_thread_lock=True, share=shared.args.share, server_name=shared.args.listen_host or '0.0.0.0', server_port=shared.args.listen_port, inbrowser=shared.args.auto_launch, auth=auth)
+        shared.gradio['interface'].launch(share=shared.args.share, server_name=shared.args.listen_host or '0.0.0.0', server_port=shared.args.listen_port, inbrowser=shared.args.auto_launch, auth=auth)
     else:
-        shared.gradio['interface'].launch(prevent_thread_lock=True, share=shared.args.share, server_port=shared.args.listen_port, inbrowser=shared.args.auto_launch, auth=auth)
+        shared.gradio['interface'].launch(share=shared.args.share, server_port=shared.args.listen_port, inbrowser=shared.args.auto_launch, auth=auth)
+    logging.info('end launch')
 
 
 if __name__ == "__main__":
@@ -873,20 +875,12 @@ if __name__ == "__main__":
         #    add_lora_to_model(shared.args.lora)
 
     # Force a character to be loaded
-    if shared.is_chat():
-        shared.persistent_interface_state.update({
-            'mode': shared.settings['mode'],
-            'character_menu': shared.args.character or shared.settings['character'],
-            'instruction_template': shared.settings['instruction_template']
-        })
-
-    print("start create_interface")
+    # if shared.is_chat():
+    #     shared.persistent_interface_state.update({
+    #         'mode': shared.settings['mode'],
+    #         'character_menu': shared.args.character or shared.settings['character'],
+    #         'instruction_template': shared.settings['instruction_template']
+    #     })
+    logging.info('start create_interface')
     # Launch the web UI
-    create_interface()
-    while True:
-        time.sleep(0.5)
-        if shared.need_restart:
-            shared.need_restart = False
-            shared.gradio['interface'].close()
-            time.sleep(0.5)
-            create_interface()
+    create_interface()