gpt4free

公开

关注 0 Fork 0 Star 1

XFEstudio/gpt4free

Code review

222ee294

hlohaus <983577+hlohaus@users.noreply.github.com>

提交于 2025-03-27 10:19

父提交 46d0b87

2 个文件 +5 -5

Modified g4f/Provider/hf/HuggingFaceMedia.py +1 -1

@@ -202,7 +202,7 @@ class HuggingFaceMedia(AsyncGeneratorProvider, ProviderModelMixin):

202 202 return provider_info, chunk

203 203 result = await response.json()

204 204 if "video" in result:

205

                            return provider_info, VideoResponse(result.get("video").get("url", result.get("video").get("url")), prompt)#video_url

205

                            return provider_info, VideoResponse(result.get("video").get("url", result.get("video").get("video_url")), prompt)

206 206 elif task == "text-to-image":

207 207

                            return provider_info, ImageResponse([item["url"] for item in result.get("images", result.get("data"))], prompt)

208 208 elif task == "text-to-video":

Modified g4f/image/copy_images.py +4 -4

@@ -72,13 +72,13 @@ async def save_response_media(response: StreamResponse, prompt: str, tags: list[

72 72 yield ImageResponse(media_url, prompt)

73 73

74 74 def get_filename(tags: list[str], alt: str, extension: str, image: str) -> str:

75 return secure_filename("".join((

75 return "".join((

76 76 f"{int(time.time())}_",

77 (f"{'_'.join([tag for tag in tags if tag])}_" if tags else ""),

78 (f"{alt}_" if alt else ""),

77 secure_filename(f"{'_'.join([tag for tag in tags if tag])}_" if tags else ""),

78 secure_filename(f"{alt}_" if alt else ""),

79 79 f"{hashlib.sha256(image.encode()).hexdigest()[:16]}",

80 80 f"{extension}"

81 )))

81 ))

82 82

83 83 async def copy_media(

84 84 images: list[str],