BlueDice
/

Katakuri-6b-torch

Inference Endpoints

Model card Files Files and versions Community

BlueDice commited on May 30, 2023

Commit

d33d685

·

1 Parent(s): b65f38d

Update handler.py

Files changed (1) hide show

handler.py +2 -1

handler.py CHANGED Viewed

@@ -53,5 +53,6 @@ class EndpointHandler():
         except Exception: pass
         return {
             "role": "AI",
-            "message": decoded_output
         }

         except Exception: pass
         return {
             "role": "AI",
+            "message": decoded_output,
+            "prompt": prompt
         }