Spaces:
Sleeping
Sleeping
Update main.py
Browse files
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__":
|