stefan-insilico
commited on
Update handler.py
Browse files- handler.py +1 -1
handler.py
CHANGED
@@ -240,6 +240,6 @@ class EndpointHandler:
|
|
240 |
except Exception as e:
|
241 |
print(e)
|
242 |
outputs, next_token_generation = [None], [None]
|
243 |
-
out = {"output": outputs, "mode": mode, 'message': f"{e}"}
|
244 |
|
245 |
return out
|
|
|
240 |
except Exception as e:
|
241 |
print(e)
|
242 |
outputs, next_token_generation = [None], [None]
|
243 |
+
out = {"output": outputs, "mode": mode, 'message': f"{e}", "input": prompt}
|
244 |
|
245 |
return out
|