emreyoleridev commited on
Commit
e5eeeb5
·
verified ·
1 Parent(s): c2c0748

monitor result

Browse files
Files changed (1) hide show
  1. src/api.py +38 -3
src/api.py CHANGED
@@ -212,10 +212,45 @@ async def grade_writing(request: WritingRequest):
212
 
213
  out = model(final_input, truncation=True, max_length=512)
214
 
215
- # sometimes out = [[{label,score}...]]; sometimes out = [{label,score}...]
216
- results = out[0] if (isinstance(out, list) and len(out) and isinstance(out[0], list)) else out
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
217
 
218
- raw = {item["label"]: item["score"] for item in results}
 
 
 
219
 
220
  def r(x): return round(x * 2) / 2
221
 
 
212
 
213
  out = model(final_input, truncation=True, max_length=512)
214
 
215
+ print("\n" + "="*80)
216
+ print("PIPELINE RAW OUTPUT")
217
+ print("="*80)
218
+
219
+ print("TYPE(out):", type(out))
220
+ print("OUT:", out)
221
+
222
+ if isinstance(out, list):
223
+ print("\nLEN(out):", len(out))
224
+ for i, item in enumerate(out):
225
+ print(f"\n--- out[{i}] ---")
226
+ print("TYPE:", type(item))
227
+ print("VALUE:", item)
228
+
229
+ if isinstance(item, list):
230
+ print(" LEN(out[%d]): %d" % (i, len(item)))
231
+ for j, sub in enumerate(item):
232
+ print(f" >>> out[{i}][{j}] TYPE:", type(sub))
233
+ print(f" >>> out[{i}][{j}] VALUE:", sub)
234
+
235
+ elif isinstance(item, dict):
236
+ print(" KEYS:", item.keys())
237
+
238
+ elif isinstance(item, str):
239
+ print(" STRING LENGTH:", len(item))
240
+
241
+ elif isinstance(out, dict):
242
+ print("DICT KEYS:", out.keys())
243
+
244
+ elif isinstance(out, str):
245
+ print("STRING LENGTH:", len(out))
246
+
247
+ print("="*80 + "\n")
248
+
249
 
250
+ return
251
+
252
+
253
+ raw = {item['label']: item['score'] for item in results}
254
 
255
  def r(x): return round(x * 2) / 2
256