Update model

gabrielmbmb · Sep 10, 2024 · decc033 · decc033
1 parent 5c41161
commit decc033
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/candle-holder-examples/examples/fill_mask_pipeline/main.rs b/candle-holder-examples/examples/fill_mask_pipeline/main.rs
@@ -12,7 +12,7 @@ fn main() -> Result<()> {
     let device = get_device_from_args()?;
     println!("Device: {:?}", device);
 
-    let pipeline = FillMaskPipeline::new("gabrielmbmb/bert-test", &device, None, None)?;
+    let pipeline = FillMaskPipeline::new("google-bert/bert-base-uncased", &device, None, None)?;
 
     let results = pipeline.run("Paris is the [MASK] of France", None)?;
     println!("`pipeline.run` results: {:?}", results);