bobmaster/matrix_chatgpt_bot
1
0
Fork 0
Code Issues 5 Pull requests 2 Projects Releases 10 Packages Wiki Activity

Remove image size limit

Browse source
This commit is contained in:
hibobmaster 2023-03-28 20:40:33 +08:00
parent 6f1420f484
commit 32ea08507d
Signed by: bobmaster
SSH key fingerprint: SHA256:5ZYgd8fg+PcNZNy4SzcSKu5JtqZyBF8kUhY7/k2viDk
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
BingImageGen.py
View file

@ -75,8 +75,10 @@ class ImageGen:
# Use regex to search for src="" # Use regex to search for src=""
image_links = regex.findall(r'src="([^"]+)"', response.text) image_links = regex.findall(r'src="([^"]+)"', response.text)
# Remove size limit
normal_image_links = [link.split("?w=")[0] for link in image_links]
# Remove duplicates # Remove duplicates
return list(set(image_links)) return list(set(normal_image_links))
def save_images(self, links: list, output_dir: str) -> str: def save_images(self, links: list, output_dir: str) -> str:
""" """