Spaces:

atlasia
/

Masked-LM-Moroccan-Darija

Running on Zero

nouamanetazi HF staff commited on 5 days ago

Commit

2d8b692

verified ·

1 Parent(s): 8113c0b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ def load_model():
     print("[INFO] Loading model... This may take a minute on Spaces")
     pipe = pipeline(
         task="fill-mask",
-        model="atlasia/XLM-RoBERTa-Morocco",
         token=key,
         device=0,
         torch_dtype=torch.float16  # Use half precision

     print("[INFO] Loading model... This may take a minute on Spaces")
     pipe = pipeline(
         task="fill-mask",
+        model="BounharAbdelaziz/XLM-RoBERTa-Morocco",
         token=key,
         device=0,
         torch_dtype=torch.float16  # Use half precision