Spaces:
Running
on
Zero
Running
on
Zero
change version of model
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ key=os.environ["HF_KEY"]
|
|
9 |
|
10 |
|
11 |
def load_model():
|
12 |
-
pipe=pipeline(task="fill-mask",model="
|
13 |
return pipe
|
14 |
|
15 |
print("[INFO] load model ...")
|
|
|
9 |
|
10 |
|
11 |
def load_model():
|
12 |
+
pipe=pipeline(task="fill-mask",model="BounharAbdelaziz/XLM-RoBERTa-Morocco",token=key,device=0)
|
13 |
return pipe
|
14 |
|
15 |
print("[INFO] load model ...")
|