返回提交历史
Modified
g4f/client/client.py
+5
-2
XFEstudio/gpt4free
Update client.py
fea056dd
代码差异
1 个文件
+5
-2
@@ -125,9 +125,12 @@ def iter_image_response(response: Iterator) -> Union[ImagesResponse, None]:
125
125
return ImagesResponse([Image(image) for image in chunk.get_list()])
126
126
127
127
def create_image(client: Client, provider: ProviderType, prompt: str, model: str = "", **kwargs) -> Iterator:
128
prompt = f"create a image with: {prompt}"
129
if provider.__name__ == "You":
128
129
130
if isinstance(provider, type) and provider.__name__ == "You":
130
131
kwargs["chat_mode"] = "create"
132
else:
133
prompt = f"create a image with: {prompt}"
131
134
return provider.create_completion(
132
135
model,
133
136
[{"role": "user", "content": prompt}],