From c9e78264166f17f6d53de9ee60aee4a617554f48 Mon Sep 17 00:00:00 2001
From: Starbeamrainbowlabs <sbrl@starbeamrainbowlabs.com>
Date: Tue, 15 Sep 2020 19:18:20 +0100
Subject: [PATCH] FilmPredictor: Fixed prediction error

---
 src/lib/FilmPredictor.mjs | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/lib/FilmPredictor.mjs b/src/lib/FilmPredictor.mjs
index 30d5eef..4a653fb 100644
--- a/src/lib/FilmPredictor.mjs
+++ b/src/lib/FilmPredictor.mjs
@@ -131,7 +131,7 @@ class FilmPredictor {
 			);
 			return image.reshape([1, ...image.shape ]);
 		})
-		let result_array = await this.model.predict(imagetensor).array();
+		let result_array = (await this.model.predict(imagetensor).array())[0];
 		return this.array2genres(result_array);
 	}
 	
@@ -140,7 +140,6 @@ class FilmPredictor {
 		for(let i = 0; i < arr.length; i++) {
 			if(arr[i] < 0.5)
 				continue;
-			
 			result.push(genres[i]);
 		}
 		return result;