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

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +4 -1
handler.py CHANGED
@@ -33,5 +33,8 @@ class EndpointHandler():
33
  )
34
 
35
  def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
36
- res = self.generate_text("Explain to me the difference between nuclear fission and fusion.")
 
 
 
37
  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