Update main.py

main.py

@@ -157,14 +157,14 @@ def generate_textAsync(request: Request, input_data: InputDataAsync):
     result_data = asyncio.run(GeneraTestoAsync("https://matteoscript-fastapi.hf.space/Genera", input_data))
     return {"response": result_data}
 
-async def make_request(session, token, data, url, index, max_retries=3):
+async def make_request(session, token, data, url, i, max_retries=3):
     headers = {
         'Content-Type': 'application/json',
         'Authorization': 'Bearer ' + token
     }
-    if index % 3 == 1:
+    if (i+1) % 3 == 1:
         data.max_new_tokens = data.max_new_tokens
-    elif index % 3 == 2:
+    elif (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 
@@ -286,7 +286,7 @@ async def GeneraTestoAsync(url, input_data):
         for data in ListaInput:
             LoggaTesto("RICHIESTA ASINCRONA", data)
             index = 0
-            tasks.extend([make_request(session, token, data, index+1, url) for index in range(input_data.NumeroGenerazioni)])
+            tasks.extend([make_request(session, token, data, index, url) for index in range(input_data.NumeroGenerazioni)])
         return await asyncio.gather(*tasks)