返回提交历史
Modified
g4f/Provider/hf_space/G4F.py
+9
-6
XFEstudio/gpt4free
Check for exception in generate of G4F
cbd66351
代码差异
1 个文件
+9
-6
@@ -83,11 +83,14 @@ class G4F(Janus_Pro_7B):
83
83
"x-zerogpu-uuid": zerogpu_uuid,
84
84
}
85
85
async def generate():
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)
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}")
91
94
background_tasks = set()
92
95
started = time.time()
93
96
task = asyncio.create_task(generate())
@@ -95,6 +98,6 @@ class G4F(Janus_Pro_7B):
95
98
task.add_done_callback(background_tasks.discard)
96
99
while background_tasks:
97
100
yield Reasoning(status=f"Generating {time.time() - started:.2f}s")
98
await asyncio.sleep(0.2)
101
await asyncio.sleep(0.5)
99
102
yield await task
100
103
yield Reasoning(status=f"Finished {time.time() - started:.2f}s")