eliott/ort
1
0
Fork 0
mirror of https://github.com/pykeio/ort synced 2026-04-25 16:34:55 +02:00
Code Issues Packages Projects Releases Wiki Activity

chore: upgrade ureq to v3.0

Browse Source
This commit is contained in:
Carson M.
2025-02-13 18:19:12 -06:00
parent e9367cca4b
commit cae6ea847a
5 changed files with 34 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
tests/squeezenet.rs
View File

@@ -1,8 +1,7 @@
use std::{
fs,
io::{self, BufRead, BufReader},
path::Path,
time::Duration
path::Path
};
use image::{ImageBuffer, Pixel, Rgb, imageops::FilterType};
@@ -101,21 +100,13 @@ fn get_imagenet_labels() -> ort::Result<Vec<String>> {
if !labels_path.exists() {
let url = "https://s3.amazonaws.com/onnx-model-zoo/synset.txt";
println!("Downloading {:?} to {:?}...", url, labels_path);
let resp = ureq::get(url)
.timeout(Duration::from_secs(180)) // 3 minutes
.call()
.map_err(Error::wrap)?;
let resp = ureq::get(url).call().map_err(Error::wrap)?;
assert!(resp.has("Content-Length"));
let len = resp.header("Content-Length").and_then(|s| s.parse::<usize>().ok()).unwrap();
println!("Downloading {} bytes...", len);
let mut reader = resp.into_reader();
let mut reader = resp.into_body().into_reader();
let f = fs::File::create(&labels_path).unwrap();
let mut writer = io::BufWriter::new(f);
let bytes_io_count = io::copy(&mut reader, &mut writer).unwrap();
assert_eq!(bytes_io_count, len as u64);
io::copy(&mut reader, &mut writer).unwrap();
}
let file = BufReader::new(fs::File::open(labels_path).unwrap());