Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -199,10 +199,9 @@ async def make_request(session, token, data, url, index, semaphore, max_retries=
|
|
199 |
data['max_new_tokens'] = max(200, data['max_new_tokens'] - 200)
|
200 |
else:
|
201 |
data['max_new_tokens'] = data['max_new_tokens'] + 200
|
202 |
-
timeout = aiohttp.ClientTimeout(total=1000)
|
203 |
for _ in range(max_retries):
|
204 |
try:
|
205 |
-
async with session.post(url, headers=headers, json=data
|
206 |
response.raise_for_status()
|
207 |
try:
|
208 |
result_data = await response.json()
|
|
|
199 |
data['max_new_tokens'] = max(200, data['max_new_tokens'] - 200)
|
200 |
else:
|
201 |
data['max_new_tokens'] = data['max_new_tokens'] + 200
|
|
|
202 |
for _ in range(max_retries):
|
203 |
try:
|
204 |
+
async with session.post(url, headers=headers, json=data) as response:
|
205 |
response.raise_for_status()
|
206 |
try:
|
207 |
result_data = await response.json()
|