Update main.py

main.py

@@ -162,9 +162,9 @@ async def make_request(session, token, data, url, i, max_retries=3):
         'Content-Type': 'application/json',
         'Authorization': 'Bearer ' + token
     }
-    if (i+1) % 3 == 1:
+    if (int(i)+1) % 3 == 1:
         data.max_new_tokens = data.max_new_tokens
-    elif (i+1) % 3 == 2:
+    elif (int(i)+1) % 3 == 2:
         data.max_new_tokens = max(200, data.max_new_tokens - 200)
     else:
         data.max_new_tokens = data.max_new_tokens + 200