hughtayloe commited on
Commit
142f207
1 Parent(s): b97a340

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -17,11 +17,11 @@ class EndpointHandler():
17
  )
18
  self.processor = AutoProcessor.from_pretrained(model_id)
19
 
20
- def __call__(self, data: Dict[Any, Any]):
21
  parameters = data.pop("inputs", data)
22
  prompt = data.pop("prompt", data)
23
-
24
  outputs = []
 
25
  for link in parameters:
26
  try:
27
  # Fetch image from URL
 
17
  )
18
  self.processor = AutoProcessor.from_pretrained(model_id)
19
 
20
+ def __call__(self, data: Dict[list, Any]):
21
  parameters = data.pop("inputs", data)
22
  prompt = data.pop("prompt", data)
 
23
  outputs = []
24
+ print(parameters)
25
  for link in parameters:
26
  try:
27
  # Fetch image from URL