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;