Changed to 1B

app.py

@@ -13,7 +13,7 @@ from datetime import datetime
 
 class LlamaChat:
     def __init__(self):
-        self.model_name = "meta-llama/Llama-3.2-3B-Instruct"
+        self.model_name = "meta-llama/Llama-3.2-1B-Instruct"
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
         self.tokenizer = None
         self.model = None