albhu commited on
Commit
17220dc
·
verified ·
1 Parent(s): 6be2107

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -12,8 +12,8 @@ if st.checkbox("Korábbi Beszélgetések Törlése"):
12
  st.session_state.previous_conversations = []
13
 
14
  # Tokenizáló modell és válaszgeneráló modell választása
15
- tokenizer_name = "joeddav/xlm-roberta-large-xnli-anli"
16
- model_name = "joeddav/xlm-roberta-large-xnli-anli"
17
 
18
  tokenizer = AutoTokenizer.from_pretrained(tokenizer_name)
19
  model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
 
12
  st.session_state.previous_conversations = []
13
 
14
  # Tokenizáló modell és válaszgeneráló modell választása
15
+ tokenizer_name = "pile-of-law/legalbert-large-1.7M-2"
16
+ model_name = "AdaptLLM/law-chat"
17
 
18
  tokenizer = AutoTokenizer.from_pretrained(tokenizer_name)
19
  model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)