rust/sbv2-api
1
0
Fork 0
mirror of https://github.com/neodyland/sbv2-api.git synced 2026-01-11 00:42:57 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix compile

Browse Source
This commit is contained in:
tuna2134
2024-10-18 13:35:23 +00:00
parent c4005808bd
commit 45a671cf52
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
sbv2_core/src/bert.rs
View File

@@ -1,5 +1,5 @@
use crate::error::Result;
use ndarray::{Array2, Ix3};
use ndarray::{Array2, Ix2};
use ort::Session;
pub fn predict(
@@ -16,7 +16,7 @@ pub fn predict(
let output = outputs["output"]
.try_extract_tensor::<f32>()?
.into_dimensionality::<Ix3>()?
.into_dimensionality::<Ix2>()?
.to_owned();
Ok(output)