Spaces:
Sleeping
Sleeping
fix error
Browse files- 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 |
-
|
| 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 |
|