Update app.py

app.py

@@ -145,8 +145,8 @@ def initialize_models():
         if llm_model is None:
             logger.info("Loading DialoGPT-small LLM...")
             try:
-                model_name = "google/flan-t5-base"
-                
+                #model_name = "google/flan-t5-base"
+                model_name = "google/gemma-4-31B"
                 llm_tokenizer = AutoTokenizer.from_pretrained(
                     model_name,
                     token=HF_TOKEN,