albhu commited on
Commit
dff56cb
·
verified ·
1 Parent(s): 87369b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ if st.checkbox("Korábbi Beszélgetések Törlése"):
15
 
16
  # Tokenizáló modell és válaszgeneráló modell választása
17
  tokenizer = AutoTokenizer.from_pretrained("pile-of-law/legalbert-large-1.7M-2")
18
- model = AutoModelForCausalLM.from_pretrained("google/gemma-7b", trust_remote_code=True)
19
 
20
  # Dokumentumfeldolgozó függvény
21
  def process_document(document_file):
 
15
 
16
  # Tokenizáló modell és válaszgeneráló modell választása
17
  tokenizer = AutoTokenizer.from_pretrained("pile-of-law/legalbert-large-1.7M-2")
18
+ model = AutoModelForCausalLM.from_pretrained("openai-community/gpt2", trust_remote_code=True)
19
 
20
  # Dokumentumfeldolgozó függvény
21
  def process_document(document_file):