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__":
|