Update app.py

app.py

@@ -3,7 +3,8 @@ from langchain_core.prompts import ChatPromptTemplate, HumanMessagePromptTemplat
 from langchain.schema import SystemMessage
 import streamlit as st
 import torch
-from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline, LlamaTokenizer
+from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
+from transformers.models.llama import LlamaTokenizer
 from langchain_community.llms.huggingface_pipeline import HuggingFacePipeline
 import nltk
 import json
@@ -35,7 +36,7 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 
 # Model selection
 model_name = st.selectbox('Choose a model:', [
-    'meta-llama/Llama-3.2-11B-Vision-Instruct',
+    'meta-llama/Llama-3.2-3B-Instruct',
     'speakleash/Bielik-11B-v2.3-Instruct',  
     # Add your private model here
 ])