XFE Git
XFE Studio Git
Git 首页 全局搜索
XFE 主站 文档 NuGet
公开
关注 0 Fork 0 Star 1
返回提交历史

XFEstudio/gpt4free

Check for exception in generate of G4F

fb58feae
hlohaus <983577+hlohaus@users.noreply.github.com>
提交于

代码差异

2 个文件 +7 -8
Modified g4f/Provider/hf_space/G4F.py +5 -8
@@ -83,14 +83,11 @@ class G4F(Janus_Pro_7B):
83 83 "x-zerogpu-uuid": zerogpu_uuid,
84 84 }
85 85 async def generate():
86 try:
87 async with session.post(cls.url_flux, json=payload, proxy=proxy, headers=headers) as response:
88 await raise_for_status(response)
89 response_data = await response.json()
90 image_url = response_data["data"][0]['url']
91 return ImageResponse(images=[image_url], alt=prompt)
92 except Exception as e:
93 return Reasoning(status=f"Error: {e.__class__.__name__}: {e}")
86 async with session.post(cls.url_flux, json=payload, proxy=proxy, headers=headers) as response:
87 await raise_for_status(response)
88 response_data = await response.json()
89 image_url = response_data["data"][0]['url']
90 return ImageResponse(images=[image_url], alt=prompt)
94 91 background_tasks = set()
95 92 started = time.time()
96 93 task = asyncio.create_task(generate())
Modified g4f/gui/server/backend_api.py +2 -0
@@ -147,6 +147,8 @@ class Backend_Api(Api):
147 147 limiter_check()
148 148 if "images" in json_data:
149 149 kwargs["images"] = json_data["images"]
150 if "zerogpu_token" in json_data:
151 kwargs["zerogpu_token"] = json_data["zerogpu_token"]
150 152 kwargs = self._prepare_conversation_kwargs(json_data, kwargs)
151 153 return self.app.response_class(
152 154 self._create_response_stream(