vivekjoshi-bluebash commited on
Commit
f2420c5
1 Parent(s): b290b73

Minor fixes

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. requirements.txt +8 -8
app.py CHANGED
@@ -11,7 +11,7 @@ from langchain_core.output_parsers import StrOutputParser
11
  openai_key = st.sidebar.text_input("Enter your OpenAI Key", type="password")
12
  # Use a pipeline as a high-level helper
13
 
14
- pipe = pipeline("image-to-text", model="nlpconnect/vit-gpt2-image-captioning")
15
  pipe2 = pipeline("text-to-speech", model="facebook/mms-tts-eng")
16
  output_parser = StrOutputParser()
17
 
 
11
  openai_key = st.sidebar.text_input("Enter your OpenAI Key", type="password")
12
  # Use a pipeline as a high-level helper
13
 
14
+ pipe = pipeline("image-to-text", model="Salesforce/blip-image-captioning-base")
15
  pipe2 = pipeline("text-to-speech", model="facebook/mms-tts-eng")
16
  output_parser = StrOutputParser()
17
 
requirements.txt CHANGED
@@ -2,12 +2,12 @@ streamlit==1.31.1
2
  pillow==10.2.0
3
  transformers==4.37.2
4
  langchain
5
- langchain-cli
6
- langchain-community
7
- langchain-core
8
- langchain-openai
9
- langchainhub
10
- langcodes
11
  langserve
12
- numpy==1.26.4
13
- tensorflow==2.15.0
 
2
  pillow==10.2.0
3
  transformers==4.37.2
4
  langchain
5
+ langchain-cli
6
+ langchain-community
7
+ langchain-core
8
+ langchain-openai
9
+ langchainhub
10
+ langcodes
11
  langserve
12
+ langsmith
13
+ torch==2.2.0