Updated model retrieval to prevent exhausting memory

Signed-off-by: Brandyn A. White <bwhite@dappervision.com>