Masked-LM-Moroccan-Darija

Runtime error

nouamanetazi HF Staff commited on Mar 6

Commit

f69e84b

verified ·

1 Parent(s): aad18c2

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="BounharAbdelaziz/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="atlasia/XLM-RoBERTa-Morocco",
         token=key,
         device=0,
         torch_dtype=torch.float16  # Use half precision