Update handler.py
Browse files- handler.py +2 -2
handler.py
CHANGED
|
@@ -91,8 +91,8 @@ class EndpointHandler:
|
|
| 91 |
|
| 92 |
# Decode outputs
|
| 93 |
decoded_outputs = self.processor.batch_decode(outputs, skip_special_tokens=True)
|
| 94 |
-
print("[INFO],decoded_outputs")
|
| 95 |
-
print("[INFO],decoded_outputs[0]")
|
| 96 |
return {"generated_text": decoded_outputs[0]}
|
| 97 |
|
| 98 |
except Exception as e:
|
|
|
|
| 91 |
|
| 92 |
# Decode outputs
|
| 93 |
decoded_outputs = self.processor.batch_decode(outputs, skip_special_tokens=True)
|
| 94 |
+
print(f"[INFO],{decoded_outputs}")
|
| 95 |
+
print(f"[INFO],{decoded_outputs[0]}")
|
| 96 |
return {"generated_text": decoded_outputs[0]}
|
| 97 |
|
| 98 |
except Exception as e:
|