commit

src/streamlit_app.py

@@ -43,7 +43,7 @@ class SimplePDFRAG:
                 'all-MiniLM-L6-v2', cache_folder=cache_dir
             )
             st.info("Loading IBM Granite model...")
-            model_name = "ibm-granite/granite-3.0-2b-instruct"
+            model_name = "ibm-granite/granite-3-2-2b-instruct"
             self.tokenizer = AutoTokenizer.from_pretrained(model_name, cache_dir=cache_dir)
             self.granite_model = AutoModelForCausalLM.from_pretrained(
                 model_name, cache_dir=cache_dir, torch_dtype=torch.float32