Itamarl commited on
Commit
fc14f0e
·
1 Parent(s): a22dfe7

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +2 -2
handler.py CHANGED
@@ -33,8 +33,8 @@ class EndpointHandler():
33
  )
34
 
35
  def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
36
- print("iiiiiiiiii " data)
37
- inputs = data.pop("inputs",data)
38
  print(inputs)
39
  res = self.generate_text("Explain to me the difference between nuclear fission and fusion." , max_length= 60)
40
  return res
 
33
  )
34
 
35
  def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
36
+ print("iiiiiiiiii ", data)
37
+ inputs = data.pop("inputs ", data)
38
  print(inputs)
39
  res = self.generate_text("Explain to me the difference between nuclear fission and fusion." , max_length= 60)
40
  return res