Update app.py

app.py

@@ -1,6 +1,5 @@
 import os
-from langchain import HuggingFacePipeline
-from langchain_core.prompts import ChatPromptTemplate, HumanMessagePromptTemplate
+from langchain import ChatPromptTemplate, HumanMessagePromptTemplate, SystemMessage
 from langchain.schema import SystemMessage
 import streamlit as st
 import torch
@@ -36,11 +35,9 @@ HF_TOKEN = os.getenv("HF_TOKEN")
 
 # Model selection
 model_name = st.selectbox('Choose a model:', [
-    'mistralai/Mistral-7B-Instruct-v0.1',
-    'meta-llama/Meta-Llama-3–8B',
-    'microsoft/Phi-3-mini-4k-instruct',
-    'microsoft/phi-1_5',
-    'speakleash/Bielik-11B-v2.3-Instruct',  # Add your private model here
+    'meta-llama/Llama-3.2-11B-Vision-Instruct',
+    'speakleash/Bielik-11B-v2.3-Instruct',  
+    # Add your private model here
 ])
 
 # Upload dataset