diff --git a/extensions/sd_api_pictures/script.py b/extensions/sd_api_pictures/script.py index b6854a02..b9fba2b9 100644 --- a/extensions/sd_api_pictures/script.py +++ b/extensions/sd_api_pictures/script.py @@ -1,14 +1,11 @@ -import json import base64 -import requests import io - -from io import BytesIO -from PIL import Image, PngImagePlugin from pathlib import Path import gradio as gr +import requests import torch +from PIL import Image import modules.chat as chat import modules.shared as shared @@ -183,4 +180,4 @@ def ui(): force_btn.click(force_pic) generate_now_btn.click(force_pic) - generate_now_btn.click(eval('chat.cai_chatbot_wrapper'), shared.input_params, shared.gradio['display'], show_progress=shared.args.no_stream) \ No newline at end of file + generate_now_btn.click(eval('chat.cai_chatbot_wrapper'), shared.input_params, shared.gradio['display'], show_progress=shared.args.no_stream)