Spaces:
Build error
Build error
update path to model
Browse files
app.py
CHANGED
|
@@ -2,10 +2,10 @@ import gradio as gr
|
|
| 2 |
from transformers import pipeline, AutoTokenizer, AutoModelForMaskedLM
|
| 3 |
|
| 4 |
# Завантаження додатково навченої моделі
|
| 5 |
-
model_path = '
|
| 6 |
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
| 7 |
model = AutoModelForMaskedLM.from_pretrained(model_path)
|
| 8 |
-
|
| 9 |
# Ініціалізація пайплайну для завдання fill-mask
|
| 10 |
fill_mask_pipeline = pipeline("fill-mask", model=model, tokenizer=tokenizer)
|
| 11 |
|
|
|
|
| 2 |
from transformers import pipeline, AutoTokenizer, AutoModelForMaskedLM
|
| 3 |
|
| 4 |
# Завантаження додатково навченої моделі
|
| 5 |
+
model_path = 'POMAHSLS/BERT_Finetunned_LawTextes' # Переконайтеся, що шлях до моделі правильний
|
| 6 |
tokenizer = AutoTokenizer.from_pretrained(model_path)
|
| 7 |
model = AutoModelForMaskedLM.from_pretrained(model_path)
|
| 8 |
+
|
| 9 |
# Ініціалізація пайплайну для завдання fill-mask
|
| 10 |
fill_mask_pipeline = pipeline("fill-mask", model=model, tokenizer=tokenizer)
|
| 11 |
|