diff --git a/extensions/send_pictures/script.py b/extensions/send_pictures/script.py index b6305bd..d2401df 100644 --- a/extensions/send_pictures/script.py +++ b/extensions/send_pictures/script.py @@ -2,12 +2,11 @@ import base64 from io import BytesIO import gradio as gr -import modules.chat as chat -import modules.shared as shared import torch -from PIL import Image from transformers import BlipForConditionalGeneration, BlipProcessor +from modules import chat, shared + # If 'state' is True, will hijack the next chat generation with # custom input text given by 'value' in the format [text, visible_text] input_hijack = {