kevinconka commited on
Commit
48379e0
1 Parent(s): e85cd1e

Fix memory allocation issue in get_model function

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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