Spaces:
Sleeping
Sleeping
kevinconka
commited on
Commit
•
48379e0
1
Parent(s):
e85cd1e
Fix memory allocation issue in get_model function
Browse files
app.py
CHANGED
@@ -56,6 +56,7 @@ def get_model():
|
|
56 |
def inference(image):
|
57 |
"""Run inference on image and return annotated image."""
|
58 |
model = get_model()
|
|
|
59 |
results = model(image)
|
60 |
return results.draw(image, diameter=4)
|
61 |
|
|
|
56 |
def inference(image):
|
57 |
"""Run inference on image and return annotated image."""
|
58 |
model = get_model()
|
59 |
+
model.model.alloc_arrays.fill(0)
|
60 |
results = model(image)
|
61 |
return results.draw(image, diameter=4)
|
62 |
|