asterixix commited on
Commit
8205bd6
1 Parent(s): db45cba

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -7
app.py CHANGED
@@ -1,6 +1,5 @@
1
  import os
2
- from langchain import HuggingFacePipeline
3
- from langchain_core.prompts import ChatPromptTemplate, HumanMessagePromptTemplate
4
  from langchain.schema import SystemMessage
5
  import streamlit as st
6
  import torch
@@ -36,11 +35,9 @@ HF_TOKEN = os.getenv("HF_TOKEN")
36
 
37
  # Model selection
38
  model_name = st.selectbox('Choose a model:', [
39
- 'mistralai/Mistral-7B-Instruct-v0.1',
40
- 'meta-llama/Meta-Llama-3–8B',
41
- 'microsoft/Phi-3-mini-4k-instruct',
42
- 'microsoft/phi-1_5',
43
- 'speakleash/Bielik-11B-v2.3-Instruct', # Add your private model here
44
  ])
45
 
46
  # Upload dataset
 
1
  import os
2
+ from langchain import ChatPromptTemplate, HumanMessagePromptTemplate, SystemMessage
 
3
  from langchain.schema import SystemMessage
4
  import streamlit as st
5
  import torch
 
35
 
36
  # Model selection
37
  model_name = st.selectbox('Choose a model:', [
38
+ 'meta-llama/Llama-3.2-11B-Vision-Instruct',
39
+ 'speakleash/Bielik-11B-v2.3-Instruct',
40
+ # Add your private model here
 
 
41
  ])
42
 
43
  # Upload dataset