MatteoScript commited on
Commit
739823d
·
verified ·
1 Parent(s): 527b418

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -2
main.py CHANGED
@@ -13,7 +13,8 @@ from enum import Enum
13
  import random
14
  import aiohttp
15
  import asyncio
16
- import json
 
17
 
18
  #--------------------------------------------------- Definizione Server FAST API ------------------------------------------------------
19
  app = FastAPI()
@@ -330,7 +331,8 @@ async def GeneraTestoAsync(url, input_data):
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
 
13
  import random
14
  import aiohttp
15
  import asyncio
16
+ import json
17
+ from types import SimpleNamespace
18
 
19
  #--------------------------------------------------- Definizione Server FAST API ------------------------------------------------------
20
  app = FastAPI()
 
331
  await asyncio.sleep(0.1)
332
  return await asyncio.gather(*tasks)
333
 
334
+ async def generate_text_internal(datajson):
335
+ data = SimpleNamespace(**datajson)
336
  temperature = data.temperature
337
  max_new_tokens = data.max_new_tokens
338
  top_p = data.top_p