Update app.py

app.py

@@ -22,7 +22,8 @@ class ChatApp:
     def load_model():
         device = "cuda" if torch.cuda.is_available() else "cpu"
         st.info(f"Using device: {device}")
-        model_name = "amiguel/classItem-FT-llama-3-1-8b-instruct"
+        #model_name = "amiguel/classItem-FT-llama-3-1-8b-instruct"
+        model_name="meta/llama-3.2-3b-instruct"
         tokenizer = AutoTokenizer.from_pretrained(model_name)
         model = AutoModelForCausalLM.from_pretrained(
             model_name,