merge deploy

src/utils/img_utils.rs

@@ -10,19 +10,21 @@ use candle_core::{DType, Device, Tensor};
 use image::{DynamicImage, ImageBuffer, ImageReader, Rgb, RgbImage, imageops};
 
 pub fn load_image_from_url(url: &str) -> Result<DynamicImage> {
-    let response = reqwest::blocking::get(url)
-        .map_err(|e| anyhow!(format!("Failed to fetch image from url: {}", e)))?;
-    let bytes = response
-        .bytes()
-        .map_err(|e| anyhow!(format!("Failed to get image bytes: {}", e)))?;
+    tokio::task::block_in_place(|| {
+        let response = reqwest::blocking::get(url)
+            .map_err(|e| anyhow!(format!("Failed to fetch image from url: {}", e)))?;
+        let bytes = response
+            .bytes()
+            .map_err(|e| anyhow!(format!("Failed to get image bytes: {}", e)))?;
 
-    let cursor = Cursor::new(bytes);
-    let img = ImageReader::new(cursor)
-        .with_guessed_format()
-        .map_err(|e| anyhow!(format!("Failed to read image format: {}", e)))?
-        .decode()
-        .map_err(|e| anyhow!(format!("Failed to decode image: {}", e)))?;
-    Ok(img)
+        let cursor = Cursor::new(bytes);
+        let img = ImageReader::new(cursor)
+            .with_guessed_format()
+            .map_err(|e| anyhow!(format!("Failed to read image format: {}", e)))?
+            .decode()
+            .map_err(|e| anyhow!(format!("Failed to decode image: {}", e)))?;
+        Ok(img)
+    })
 }
 
 pub fn load_image_from_base64(base64_data: &str) -> Result<DynamicImage> {

src/utils/mod.rs

@@ -255,10 +255,11 @@ pub fn get_logit_processor(
     temperature: Option<f32>,
     top_p: Option<f32>,
     top_k: Option<usize>,
+    seed: u64,
 ) -> LogitsProcessor {
     match top_k {
         None => LogitsProcessor::new(
-            34562,
+            seed,
             temperature.map(|temp| temp as f64),
             top_p.map(|tp| tp as f64),
         ),
@@ -277,7 +278,7 @@ pub fn get_logit_processor(
                     },
                 },
             };
-            LogitsProcessor::from_sampling(34562, sampling)
+            LogitsProcessor::from_sampling(seed, sampling)
         }
     }
 }