MatteoScript commited on
Commit
07db5b7
·
verified ·
1 Parent(s): 07fc210

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -2
main.py CHANGED
@@ -134,7 +134,10 @@ async def make_request(session, token, data, url):
134
  'Authorization': 'Bearer ' + token
135
  }
136
  async with session.post(url, headers=headers, json=data) as response:
137
- result_data = await response.json()
 
 
 
138
  print(result_data)
139
  return result_data
140
 
@@ -155,7 +158,6 @@ async def GeneraTestoAsync(url, input_data):
155
  'instruction': input_data.instruction
156
  }
157
  tasks.extend([make_request(session, token, data, url) for _ in range(input_data.NumeroGenerazioni)])
158
- #tasks = [make_request(session, token, data, url) for _ in range(input_data.NumeroGenerazioni)]
159
  return await asyncio.gather(*tasks)
160
 
161
  #--------------------------------------------------- Generazione IMMAGINE ------------------------------------------------------
 
134
  'Authorization': 'Bearer ' + token
135
  }
136
  async with session.post(url, headers=headers, json=data) as response:
137
+ try:
138
+ result_data = await response.json()
139
+ except aiohttp.ContentTypeError:
140
+ result_data = await response.text()
141
  print(result_data)
142
  return result_data
143
 
 
158
  'instruction': input_data.instruction
159
  }
160
  tasks.extend([make_request(session, token, data, url) for _ in range(input_data.NumeroGenerazioni)])
 
161
  return await asyncio.gather(*tasks)
162
 
163
  #--------------------------------------------------- Generazione IMMAGINE ------------------------------------------------------