Ver Fonte

Minor fix

oobabooga há 2 anos atrás
pai
commit
bfb1be2820
1 ficheiros alterados com 2 adições e 2 exclusões
  1. 2 2
      extensions/send_pictures/script.py

+ 2 - 2
extensions/send_pictures/script.py

@@ -26,7 +26,7 @@ def caption_image(raw_image):
 def generate_chat_picture(picture, name1, name2):
     text = f'*{name1} sends {name2} a picture that contains the following: "{caption_image(picture)}"*'
     # lower the resolution of sent images for the chat, otherwise the log size gets out of control quickly with all the base64 values in visible history
-    image.thumbnail((300, 300))
+    picture.thumbnail((300, 300))
     buffer = BytesIO()
     picture.save(buffer, format="JPEG")
     img_str = base64.b64encode(buffer.getvalue()).decode('utf-8')
@@ -45,4 +45,4 @@ def ui():
     picture_select.upload(eval(function_call), shared.input_params, shared.gradio['display'], show_progress=shared.args.no_stream)
 
     # Clear the picture from the upload field
-    picture_select.upload(lambda : None, [], [picture_select], show_progress=False)
+    picture_select.upload(lambda : None, [], [picture_select], show_progress=False)