zliang commited on
Commit
a6e5795
·
1 Parent(s): 230f819

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,4 +1,3 @@
1
- import openai
2
  import streamlit as st
3
  from langchain.llms import OpenAI
4
  from langchain.chat_models import ChatOpenAI
@@ -20,7 +19,7 @@ import time
20
  # return docs
21
 
22
  model_name = "intfloat/e5-large-v2"
23
- model_kwargs = {'device': 'cuda'}
24
  encode_kwargs = {'normalize_embeddings': False}
25
  embeddings = HuggingFaceEmbeddings(
26
  model_name=model_name,
 
 
1
  import streamlit as st
2
  from langchain.llms import OpenAI
3
  from langchain.chat_models import ChatOpenAI
 
19
  # return docs
20
 
21
  model_name = "intfloat/e5-large-v2"
22
+ model_kwargs = {'device': 'cpu'}
23
  encode_kwargs = {'normalize_embeddings': False}
24
  embeddings = HuggingFaceEmbeddings(
25
  model_name=model_name,