Utk113Raj commited on
Commit
2b5c64d
·
verified ·
1 Parent(s): 18f1cf6

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -8,7 +8,7 @@ class EndpointHandler:
8
  def __call__(self, data):
9
  # If data is wrapped in 'inputs' (as Hugging Face does), unwrap it
10
  if isinstance(data, dict) and "inputs" in data:
11
- data = data["inputs"]
12
 
13
  text = data.get("text", "")
14
  labels = data.get("labels", [])
 
8
  def __call__(self, data):
9
  # If data is wrapped in 'inputs' (as Hugging Face does), unwrap it
10
  if isinstance(data, dict) and "inputs" in data:
11
+ data = data["inputs"]
12
 
13
  text = data.get("text", "")
14
  labels = data.get("labels", [])