rockerritesh commited on
Commit
6e3dbc0
·
verified ·
1 Parent(s): 682a089

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -2
main.py CHANGED
@@ -145,7 +145,7 @@ def predict(image):
145
  # Clean up intermediates
146
  del model_inputs, generation
147
  clean_memory()
148
-
149
  return decoded
150
 
151
  @app.post("/extract_text")
@@ -231,7 +231,6 @@ async def health_check():
231
  image = Image.fromarray(random_data)
232
  predict(image)
233
  clean_memory()
234
- del model,processor
235
  return {"status": "healthy"}
236
 
237
  # if __name__ == "__main__":
 
145
  # Clean up intermediates
146
  del model_inputs, generation
147
  clean_memory()
148
+ del model,processor
149
  return decoded
150
 
151
  @app.post("/extract_text")
 
231
  image = Image.fromarray(random_data)
232
  predict(image)
233
  clean_memory()
 
234
  return {"status": "healthy"}
235
 
236
  # if __name__ == "__main__":