Updated provider (g4f/Provider/nexra/NexraDallE.py)

This commit is contained in:
kqlio67 2024-10-22 12:35:30 +03:00
parent 5647d7db1e
commit bdf9db27a7

View file

@ -24,6 +24,7 @@ class NexraDallE(AbstractProvider, ProviderModelMixin):
cls,
model: str,
messages: Messages,
response: str = "url", # base64 or url
**kwargs
) -> CreateResult:
model = cls.get_model(model)
@ -35,7 +36,7 @@ class NexraDallE(AbstractProvider, ProviderModelMixin):
data = {
"prompt": messages[-1]["content"],
"model": model,
"response": "url"
"response": response
}
response = requests.post(cls.api_endpoint, headers=headers, json=data)