Update server.py

server.py

@@ -10,6 +10,7 @@ warnings.filterwarnings('ignore', category=UserWarning, message='TypedStorage is
 
 with RequestBlocker():
     import gradio as gr
+    from gradio import components, Progress
 
 import matplotlib
 matplotlib.use('Agg')  # This fixes LaTeX rendering on some systems
@@ -127,8 +128,8 @@ def count_tokens(text):
     except:
         return 'Couldn\'t count the number of tokens. Is a tokenizer loaded?'
 
-
-def download_model_wrapper(repo_id, progress=gr.Progress()):
+#def download_model_wrapper(repo_id, progress=gr.Progress()):
+def download_model_wrapper(repo_id, progress=Progress()):
     try:
         downloader_module = importlib.import_module("download-model")
         downloader = downloader_module.ModelDownloader()
@@ -439,7 +440,9 @@ def create_settings_menus(default_preset):
 def create_file_saving_menus():
 
     # Text file saver
-    with gr.Box(visible=False, elem_classes='file-saver') as shared.gradio['file_saver']:
+    #with gr.Box(visible=False, elem_classes='file-saver') as shared.gradio['file_saver']:
+    #with gr.Column(visible=False, elem_classes='file-saver') as shared.gradio['file_saver']:
+    with gr.Group(visible=False, elem_classes='file-saver') as shared.gradio['file_saver']:
         shared.gradio['save_filename'] = gr.Textbox(lines=1, label='File name')
         shared.gradio['save_root'] = gr.Textbox(lines=1, label='File folder', info='For reference. Unchangeable.', interactive=False)
         shared.gradio['save_contents'] = gr.Textbox(lines=10, label='File contents')
@@ -1066,14 +1069,14 @@ def create_interface():
         # Extensions block
         extensions_module.create_extensions_block()
 
-    # Launch the interface
-    shared.gradio['interface'].queue()
-    with OpenMonkeyPatch():
-        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)
-        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(
+            prevent_thread_lock=True,
+            share=False,  # Set this to True
+            server_name="0.0.0.0",  # Add this line
+            server_port=shared.args.listen_port,
+            inbrowser=shared.args.auto_launch,
+            auth=auth
+        )
 
 if __name__ == "__main__":
     # Loading custom settings