samtuckervegan commited on
Commit
f176ff5
·
verified ·
1 Parent(s): f61bc02

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +0 -5
handler.py CHANGED
@@ -18,11 +18,6 @@ class EndpointHandler:
18
 
19
  def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
20
  prompt = data.get("inputs", "")
21
- if not prompt:
22
- return {"error": "Missing 'inputs' field."}
23
-
24
- if not prompt.startswith("<|begin_of_text|>"):
25
- prompt = f"<|begin_of_text|>{prompt}"
26
 
27
  params = data.get("parameters", {})
28
  outputs = self.generator(
 
18
 
19
  def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
20
  prompt = data.get("inputs", "")
 
 
 
 
 
21
 
22
  params = data.get("parameters", {})
23
  outputs = self.generator(