Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 import torch
 
 # Load IBM Granite model
-model_id = "ibm/granite-3b-instruct-v1"
+model_id = "ibm/granite-3.3-2b-instruct"
 token = os.getenv("HF_TOKEN")  # Use Hugging Face Secrets
 tokenizer = AutoTokenizer.from_pretrained(model_id, use_auth_token=token)
 model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype=torch.float32, use_auth_token=token)