POMAHSLS commited on
Commit
d0c0e2a
·
verified ·
1 Parent(s): 79ce54b

update path to model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,10 +2,10 @@ import gradio as gr
2
  from transformers import pipeline, AutoTokenizer, AutoModelForMaskedLM
3
 
4
  # Завантаження додатково навченої моделі
5
- model_path = 'H:/FinetunningExam/ModelBert' # Переконайтеся, що шлях до моделі правильний
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