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

fix error

Browse files
Files changed (1) hide show
  1. src/api.py +4 -38
src/api.py CHANGED
@@ -210,47 +210,13 @@ async def grade_writing(request: WritingRequest):
210
  else:
211
  final_input = f"{request.prompt} [SEP] {request.essay}"
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
 
 
210
  else:
211
  final_input = f"{request.prompt} [SEP] {request.essay}"
212
 
213
+ results = model(final_input, truncation=True, max_length=512)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
214
 
215
+ print(results)
216
 
217
  raw = {item['label']: item['score'] for item in results}
218
+
219
+ print(raw)
220
 
221
  def r(x): return round(x * 2) / 2
222