oobabooga 2 лет назад
Родитель
Сommit
572f1d8bdb
1 измененных файлов с 2 добавлено и 3 удалено
  1. 2 3
      extensions/send_pictures/script.py

+ 2 - 3
extensions/send_pictures/script.py

@@ -2,12 +2,11 @@ import base64
 from io import BytesIO
 from io import BytesIO
 
 
 import gradio as gr
 import gradio as gr
-import modules.chat as chat
-import modules.shared as shared
 import torch
 import torch
-from PIL import Image
 from transformers import BlipForConditionalGeneration, BlipProcessor
 from transformers import BlipForConditionalGeneration, BlipProcessor
 
 
+from modules import chat, shared
+
 # If 'state' is True, will hijack the next chat generation with
 # If 'state' is True, will hijack the next chat generation with
 # custom input text given by 'value' in the format [text, visible_text]
 # custom input text given by 'value' in the format [text, visible_text]
 input_hijack = {
 input_hijack = {