Update app.py

app.py

@@ -63,11 +63,13 @@ def initialize_database(document, chunk_size, chunk_overlap, progress=gr.Progres
 def initialize_LLM(llm_option, llm_temperature, max_tokens, top_k, vector_db, progress=gr.Progress(), language="italiano"):
     logger.info("Initializing LLM chain...")
 
-    # Define the default LLMS based on the language
-    if language == "italian":
-        default_llm = "google/gemma-7b-it"
+    # Definizione del modello LLM
+    if language == "italiano":
+        model = AutoModelForCausalLM.from_pretrained("google/gemma-7b-it")
     else:
-        default_llm = "google/gemma-7b"  # English version
+        model = AutoModelForCausalLM.from_pretrained("google/gemma-7b")
+
+    tokenizer = AutoTokenizer.from_pretrained("google/gemma-7b-it" if language == "italiano" else "google/gemma-7b")
 
     qa_chain = ConversationalRetrievalChain.from_llm(
         llm=model,