Update main.py

main.py

@@ -163,11 +163,11 @@ async def make_request(session, token, data, url, index, max_retries=3):
         'Authorization': 'Bearer ' + token
     }
     if (int(index)+1) % 3 == 1:
-        data.max_new_tokens = data.max_new_tokens
+        data['max_new_tokens'] = data['max_new_tokens'] 
     elif (int(index)+1) % 3 == 2:
-        data.max_new_tokens = max(200, data.max_new_tokens - 200)
+        data['max_new_tokens'] = max(200, data['max_new_tokens'] - 200)
     else:
-        data.max_new_tokens = data.max_new_tokens + 200 
+        data['max_new_tokens'] = data['max_new_tokens'] + 200 
     for _ in range(max_retries):
         try:
             async with session.post(url, headers=headers, json=data) as response: