Create new API

This commit is contained in:
oobabooga
2023-04-06 00:31:58 -03:00
parent e6569653b1
commit 9c3a585915
2 changed files with 41 additions and 1 deletions

View File

@@ -15,7 +15,7 @@ import gradio as gr
from PIL import Image
import modules.extensions as extensions_module
from modules import chat, shared, training, ui
from modules import chat, shared, training, ui, api
from modules.html_generator import chat_html_wrapper
from modules.LoRA import add_lora_to_model
from modules.models import load_model, load_soft_prompt
@@ -538,6 +538,8 @@ def create_interface():
else:
shared.gradio[k].release(lambda state, value, copy=k: change_dict_value(state, copy, value), inputs=[shared.gradio['generate_state'], shared.gradio[k]], outputs=shared.gradio['generate_state'])
api.create_apis()
# Authentication
auth = None
if shared.args.gradio_auth_path is not None: