Update main.py

main.py

@@ -40,7 +40,7 @@ class InputDataAsync(BaseModel):
     top_p: float = 0.95
     repetition_penalty: float = 1.0
     NumeroGenerazioni: int = 1
-    StringaSplit: str = ''
+    StringaSplit: str = '********'
     
 class PostSpazio(BaseModel):
     nomeSpazio: str
@@ -105,15 +105,19 @@ async def make_request(session, token, data, url):
 
 async def GeneraTestoAsync(url, input_data):
     token = os.getenv('TOKEN')
-    data = {
-        'input': input_data.input,
-        'temperature': input_data.temperature,
-        'max_new_tokens': input_data.max_new_tokens,
-        'top_p': input_data.top_p,
-        'repetition_penalty': input_data.repetition_penalty
-    }
     async with aiohttp.ClientSession() as session:
-        tasks = [make_request(session, token, data, url) for _ in range(input_data.NumeroGenerazioni)]  
+        input_strings = input_data.input.split(input_data.StringaSplit)
+        tasks = []
+        for input_string in input_strings:
+            data = {
+                'input': input_string.strip(),
+                'temperature': input_data.temperature,
+                'max_new_tokens': input_data.max_new_tokens,
+                'top_p': input_data.top_p,
+                'repetition_penalty': input_data.repetition_penalty
+            }
+            tasks.append(make_request(session, token, data, url) for _ in range(input_data.NumeroGenerazioni))
+        #tasks = [make_request(session, token, data, url) for _ in range(input_data.NumeroGenerazioni)]  
         return await asyncio.gather(*tasks)
     
 #--------------------------------------------------- Generazione IMMAGINE ------------------------------------------------------