zishuo/gpt4free
1
0
Fork 0
mirror of https://github.com/xtekky/gpt4free.git synced 2025-10-20 15:06:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Check for exception in generate of G4F

Browse Source
This commit is contained in:
hlohaus
2025-02-05 19:30:47 +01:00
parent cbd663511c
commit fb58feae87
2 changed files with 7 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
g4f/Provider/hf_space/G4F.py
View File

@@ -83,14 +83,11 @@ class G4F(Janus_Pro_7B):
"x-zerogpu-uuid": zerogpu_uuid, "x-zerogpu-uuid": zerogpu_uuid,
} }
async def generate(): async def generate():
try:
async with session.post(cls.url_flux, json=payload, proxy=proxy, headers=headers) as response: async with session.post(cls.url_flux, json=payload, proxy=proxy, headers=headers) as response:
await raise_for_status(response) await raise_for_status(response)
response_data = await response.json() response_data = await response.json()
image_url = response_data["data"][0]['url'] image_url = response_data["data"][0]['url']
return ImageResponse(images=[image_url], alt=prompt) return ImageResponse(images=[image_url], alt=prompt)
except Exception as e:
return Reasoning(status=f"Error: {e.__class__.__name__}: {e}")
background_tasks = set() background_tasks = set()
started = time.time() started = time.time()
task = asyncio.create_task(generate()) task = asyncio.create_task(generate())

2
g4f/gui/server/backend_api.py
View File

@@ -147,6 +147,8 @@ class Backend_Api(Api):
limiter_check() limiter_check()
if "images" in json_data: if "images" in json_data:
kwargs["images"] = json_data["images"] kwargs["images"] = json_data["images"]
if "zerogpu_token" in json_data:
kwargs["zerogpu_token"] = json_data["zerogpu_token"]
kwargs = self._prepare_conversation_kwargs(json_data, kwargs) kwargs = self._prepare_conversation_kwargs(json_data, kwargs)
return self.app.response_class( return self.app.response_class(
self._create_response_stream( self._create_response_stream(