Update main.py
Browse files
main.py
CHANGED
|
@@ -117,8 +117,7 @@ async def submit_image_generation(prompt: str, size: str = "1024x1024") -> str:
|
|
| 117 |
async def check_generation_status(operation_id: str) -> dict:
|
| 118 |
"""Check the status of image generation"""
|
| 119 |
headers = {
|
| 120 |
-
"accept": "application/json, text/plain
|
| 121 |
-
, */*",
|
| 122 |
"authorization": f"Bearer {BEARER_TOKEN}",
|
| 123 |
"content-type": "application/json",
|
| 124 |
"origin": "https://desktop.captions.ai",
|
|
|
|
| 117 |
async def check_generation_status(operation_id: str) -> dict:
|
| 118 |
"""Check the status of image generation"""
|
| 119 |
headers = {
|
| 120 |
+
"accept": "application/json, text/plain, */*",
|
|
|
|
| 121 |
"authorization": f"Bearer {BEARER_TOKEN}",
|
| 122 |
"content-type": "application/json",
|
| 123 |
"origin": "https://desktop.captions.ai",
|