Update app.py

app.py

@@ -1,6 +1,4 @@
 import streamlit as st
-import pdfplumber
-import docx
 from transformers import AutoTokenizer, AutoModelForCausalLM
 
 st.title("Jogi Dokumentumok AI Válaszgenerálás")
@@ -14,11 +12,11 @@ if st.checkbox("Korábbi Beszélgetések Törlése"):
     st.session_state.previous_conversations = []
 
 # Tokenizáló modell és válaszgeneráló modell választása
-tokenizer_name = st.selectbox("Tokenizáló Modell", ["bert-base-uncased", "bert-base-multilingual-cased"])
-generator_name = st.selectbox("Válaszgeneráló Modell", ["bert-base-uncased", "bert-base-multilingual-cased"])
+tokenizer_name = "joeddav/xlm-roberta-large-xnli-anli"
+model_name = "joeddav/xlm-roberta-large-xnli-anli"
 
 tokenizer = AutoTokenizer.from_pretrained(tokenizer_name)
-model = AutoModelForCausalLM.from_pretrained(generator_name, trust_remote_code=True)
+model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
 
 # Definiáljuk a válaszgeneráló függvényt
 def generate_response(input_text, tokenizer, model):
@@ -32,8 +30,6 @@ def generate_response(input_text, tokenizer, model):
     response = tokenizer.decode(output[0], skip_special_tokens=True)
     return response
 
-
-
 # Dokumentum feltöltése drag and drop segítségével
 document_file = st.file_uploader("Húzd ide a dokumentumot vagy kattints a feltöltéshez", type=["pdf", "docx", "doc"])