Mirrors/text-generation-webui
1
0
Fork 0
mirror of https://github.com/oobabooga/text-generation-webui.git synced 2024-12-25 05:48:55 +01:00
Code Issues Packages Projects Releases Wiki Activity

Minor change to API code

Browse Source
This commit is contained in:
oobabooga 2023-04-14 12:11:47 -03:00
parent 581f739b2f
commit 1901d238e1
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
modules/api.py
View File

@ -5,6 +5,9 @@ import gradio as gr
from modules import shared from modules import shared
from modules.text_generation import generate_reply from modules.text_generation import generate_reply
# set this to True to rediscover the fn_index using the browser DevTools
VISIBLE = False
def generate_reply_wrapper(string): def generate_reply_wrapper(string):
generate_params = { generate_params = {
@ -30,9 +33,9 @@ def generate_reply_wrapper(string):
def create_apis(): def create_apis():
t1 = gr.Textbox(visible=False) t1 = gr.Textbox(visible=VISIBLE)
t2 = gr.Textbox(visible=False) t2 = gr.Textbox(visible=VISIBLE)
dummy = gr.Button(visible=False) dummy = gr.Button(visible=VISIBLE)
input_params = [t1] input_params = [t1]
output_params = [t2] + [shared.gradio[k] for k in ['markdown', 'html']] output_params = [t2] + [shared.gradio[k] for k in ['markdown', 'html']]