JeremyArancio
/

llm-tolkien

Inference Endpoints

Model card Files Files and versions Community

JeremyArancio commited on Jun 1, 2023

Commit

f85d258

•

1 Parent(s): adf79f2

Update handler

Files changed (1) hide show

handler.py +1 -1

handler.py CHANGED Viewed

@@ -24,7 +24,7 @@ class EndpointHandler():
         """
         LOGGER.info(f"Received data: {data}")
         # Get inputs
-        prompt = data.pop("prompt", None)
         parameters = data.pop("parameters", None)
         if prompt is None:
             raise ValueError("Missing prompt.")

         """
         LOGGER.info(f"Received data: {data}")
         # Get inputs
+        prompt = data.pop("inputs", None)
         parameters = data.pop("parameters", None)
         if prompt is None:
             raise ValueError("Missing prompt.")