add gradio version warning

        code()

    except Exception as e:

        display(task, e)

def check_versions():

    from packaging import version

    from modules import shared

    import torch

    import gradio

    expected_torch_version = "2.0.0"

    expected_xformers_version = "0.0.20"

    expected_gradio_version = "3.39.0"

    if version.parse(torch.__version__) < version.parse(expected_torch_version):

        print_error_explanation(f"""

You are running torch {torch.__version__}.

The program is tested to work with torch {expected_torch_version}.

To reinstall the desired version, run with commandline flag --reinstall-torch.

Beware that this will cause a lot of large files to be downloaded, as well as

there are reports of issues with training tab on the latest version.

Use --skip-version-check commandline argument to disable this check.

        """.strip())

    if shared.xformers_available:

        import xformers

        if version.parse(xformers.__version__) < version.parse(expected_xformers_version):

            print_error_explanation(f"""

You are running xformers {xformers.__version__}.

The program is tested to work with xformers {expected_xformers_version}.

To reinstall the desired version, run with commandline flag --reinstall-xformers.

Use --skip-version-check commandline argument to disable this check.

            """.strip())

    if gradio.__version__ != expected_gradio_version:

        print_error_explanation(f"""

You are running gradio {gradio.__version__}.

The program is designed to work with gradio {expected_gradio_version}.

Using a different version of gradio is extremely likely to break the program.

Reasons why you have the mismatched gradio version can be:

  - you use --skip-install flag.

  - you use webui.py to start the program instead of launch.py.

  - an extension installs the incompatible gradio version.

Use --skip-version-check commandline argument to disable this check.

        """.strip())

from fastapi import FastAPI

from fastapi.middleware.cors import CORSMiddleware

from fastapi.middleware.gzip import GZipMiddleware

from packaging import version

import logging

import ldm.modules.encoders.modules  # noqa: F401

startup_timer.record("import ldm")

from modules import extra_networks

from modules.call_queue import wrap_gradio_gpu_call, wrap_queued_call, queue_lock  # noqa: F401

    torch.__long_version__ = torch.__version__

    torch.__version__ = re.search(r'[\d.]+[\d]', torch.__version__).group(0)

from modules import shared

if not shared.cmd_opts.skip_version_check:

    errors.check_versions()

import modules.codeformer_model as codeformer

import modules.gfpgan_model as gfpgan

from modules import shared, sd_samplers, upscaler, extensions, localization, ui_tempdir, ui_extra_networks, config_states

from modules import sd_samplers, upscaler, extensions, localization, ui_tempdir, ui_extra_networks, config_states

import modules.face_restoration

import modules.img2img

    asyncio.set_event_loop_policy(AnyThreadEventLoopPolicy())

def check_versions():

    if shared.cmd_opts.skip_version_check:

        return

    expected_torch_version = "2.0.0"

    if version.parse(torch.__version__) < version.parse(expected_torch_version):

        errors.print_error_explanation(f"""

You are running torch {torch.__version__}.

The program is tested to work with torch {expected_torch_version}.

To reinstall the desired version, run with commandline flag --reinstall-torch.

Beware that this will cause a lot of large files to be downloaded, as well as

there are reports of issues with training tab on the latest version.

Use --skip-version-check commandline argument to disable this check.

        """.strip())

    expected_xformers_version = "0.0.20"

    if shared.xformers_available:

        import xformers

        if version.parse(xformers.__version__) < version.parse(expected_xformers_version):

            errors.print_error_explanation(f"""

You are running xformers {xformers.__version__}.

The program is tested to work with xformers {expected_xformers_version}.

To reinstall the desired version, run with commandline flag --reinstall-xformers.

Use --skip-version-check commandline argument to disable this check.

            """.strip())

def restore_config_state_file():

    config_state_file = shared.opts.restore_config_state_file

    if config_state_file == "":

    fix_asyncio_event_loop_policy()

    validate_tls_options()

    configure_sigint_handler()

    check_versions()

    modelloader.cleanup_models()

    configure_opts_onchange()