Upload 2 files

ui/globals.py

@@ -4,6 +4,7 @@ SELECTION_FACES_DATA = None
 ui_SELECTED_INPUT_FACE_INDEX = 0
 
 ui_selected_enhancer = None
+ui_upscale = None
 ui_blend_ratio = None
 ui_input_thumbs = []
 ui_target_thumbs = []

ui/main.py

@@ -27,15 +27,21 @@ def prepare_environment():
     os.environ["GRADIO_TEMP_DIR"] = os.environ["TEMP"]
     os.environ['GRADIO_ANALYTICS_ENABLED'] = '0'
 
-
 def run():
     from roop.core import decode_execution_providers, set_display_ui
 
     prepare_environment()
 
     set_display_ui(show_msg)
+    if roop.globals.CFG.provider == "cuda" and util.has_cuda_device() == False:
+       roop.globals.CFG.provider = "cpu"
+
     roop.globals.execution_providers = decode_execution_providers([roop.globals.CFG.provider])
-    print(f'Using provider {roop.globals.execution_providers} - Device:{util.get_device()}')    
+    gputype = util.get_device()
+    if gputype == 'cuda':
+        util.print_cuda_info()
+        
+    print(f'Using provider {roop.globals.execution_providers} - Device:{gputype}')
     
     run_server = True
     uii.ui_restart_server = False
@@ -57,7 +63,7 @@ def run():
         if server_port <= 0:
             server_port = None
         ssl_verify = False if server_name == '0.0.0.0' else True
-        with gr.Blocks(title=f'{roop.metadata.name} {roop.metadata.version}', theme=roop.globals.CFG.selected_theme, css=mycss) as ui:
+        with gr.Blocks(title=f'{roop.metadata.name} {roop.metadata.version}', theme=roop.globals.CFG.selected_theme, css=mycss, delete_cache=(60, 86400)) as ui:
             with gr.Row(variant='compact'):
                     gr.Markdown(f"### [{roop.metadata.name} {roop.metadata.version}](https://github.com/C0untFloyd/roop-unleashed)")
                     gr.HTML(util.create_version_html(), elem_id="versions")
@@ -66,10 +72,11 @@ def run():
             facemgr_tab()
             extras_tab()
             settings_tab()
+        launch_browser = roop.globals.CFG.launch_browser
 
         uii.ui_restart_server = False
         try:
-            ui.queue().launch(inbrowser=True, server_name=server_name, server_port=server_port, share=roop.globals.CFG.server_share, ssl_verify=ssl_verify, prevent_thread_lock=True, show_error=True)
+            ui.queue().launch(inbrowser=launch_browser, server_name=server_name, server_port=server_port, share=roop.globals.CFG.server_share, ssl_verify=ssl_verify, prevent_thread_lock=True, show_error=True)
         except Exception as e:
             print(f'Exception {e} when launching Gradio Server!')
             uii.ui_restart_server = True