From decc033a157e9bbc011c74e97b7526a8e8dcdd3c Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Gabriel=20Mart=C3=ADn=20Bl=C3=A1zquez?=
 <gmartinbdev@gmail.com>
Date: Tue, 10 Sep 2024 16:41:00 +0200
Subject: [PATCH] Update model

---
 candle-holder-examples/examples/fill_mask_pipeline/main.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/candle-holder-examples/examples/fill_mask_pipeline/main.rs b/candle-holder-examples/examples/fill_mask_pipeline/main.rs
index 1fec3c8..44b111e 100644
--- 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);