RanjithaRuttala commited on
Commit
0a11e2e
·
verified ·
1 Parent(s): f4d104e

Update handler.py

Browse files

rename generated_text to generated

Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -78,4 +78,4 @@ class EndpointHandler:
78
  )
79
 
80
  generated = generated.replace("<fim_middle>", "").replace("<fim_suffix>", "").strip()
81
- return {"generated_text": generated_text.strip()}
 
78
  )
79
 
80
  generated = generated.replace("<fim_middle>", "").replace("<fim_suffix>", "").strip()
81
+ return {"generated_text": generated.strip()}