Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-11-22 08:07:56 +01:00
Code Issues Packages Projects Releases Wiki Activity

Block the "To create a public link, set share=True" gradio message

Browse Source
This commit is contained in:
oobabooga 2024-02-25 15:06:08 -08:00
parent d34126255d
commit 4164e29416
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
modules/block_requests.py
View File

@ -7,6 +7,7 @@ from modules.logging_colors import logger
original_open = open original_open = open
original_get = requests.get original_get = requests.get
original_print = print
class RequestBlocker: class RequestBlocker:
@ -22,9 +23,11 @@ class OpenMonkeyPatch:
def __enter__(self): def __enter__(self):
builtins.open = my_open builtins.open = my_open
builtins.print = my_print
def __exit__(self, exc_type, exc_value, traceback): def __exit__(self, exc_type, exc_value, traceback):
builtins.open = original_open builtins.open = original_open
builtins.print = original_print
def my_get(url, **kwargs): def my_get(url, **kwargs):
@ -45,3 +48,15 @@ def my_open(*args, **kwargs):
return io.BytesIO(file_contents) return io.BytesIO(file_contents)
else: else:
return original_open(*args, **kwargs) return original_open(*args, **kwargs)
def my_print(*args, **kwargs):
if len(args) > 0 and 'To create a public link, set `share=True`' in args[0]:
return
else:
if len(args) > 0 and 'Running on local URL' in args[0]:
args = list(args)
args[0] = f"\n{args[0].strip()}\n"
args = tuple(args)
original_print(*args, **kwargs)