Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -325,11 +325,22 @@ async def GeneraTestoAsync(url, input_data):
|
|
325 |
ListaInput = await CreaListaInput(input_data)
|
326 |
for data in ListaInput:
|
327 |
LoggaTesto("RICHIESTA ASINCRONA", data)
|
328 |
-
tasks.extend([make_request(session, token, data, url, index, semaphore) for index in range(input_data.NumeroGenerazioni)])
|
|
|
329 |
await asyncio.sleep(0.1)
|
330 |
return await asyncio.gather(*tasks)
|
331 |
|
332 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
333 |
#--------------------------------------------------- Generazione IMMAGINE ------------------------------------------------------
|
334 |
style_image = {
|
335 |
"PROFESSIONAL-PHOTO": {
|
|
|
325 |
ListaInput = await CreaListaInput(input_data)
|
326 |
for data in ListaInput:
|
327 |
LoggaTesto("RICHIESTA ASINCRONA", data)
|
328 |
+
#tasks.extend([make_request(session, token, data, url, index, semaphore) for index in range(input_data.NumeroGenerazioni)])
|
329 |
+
tasks.extend([generate_text_internal(data) for _ in range(input_data.NumeroGenerazioni)])
|
330 |
await asyncio.sleep(0.1)
|
331 |
return await asyncio.gather(*tasks)
|
332 |
|
333 |
+
async def generate_text_internal(data):
|
334 |
+
temperature = data.temperature
|
335 |
+
max_new_tokens = data.max_new_tokens
|
336 |
+
top_p = data.top_p
|
337 |
+
repetition_penalty = data.repetition_penalty
|
338 |
+
input_text = generate_input_text(data)
|
339 |
+
max_new_tokens = min(max_new_tokens, 29500 - len(input_text))
|
340 |
+
history = []
|
341 |
+
generated_response = generate(input_text, history, temperature, max_new_tokens, top_p, repetition_penalty)
|
342 |
+
return generated_response
|
343 |
+
|
344 |
#--------------------------------------------------- Generazione IMMAGINE ------------------------------------------------------
|
345 |
style_image = {
|
346 |
"PROFESSIONAL-PHOTO": {
|