annikwag commited on
Commit
c9e65e2
·
verified ·
1 Parent(s): 8e1fb27

Update appStore/embed.py

Browse files
Files changed (1) hide show
  1. appStore/embed.py +2 -0
appStore/embed.py CHANGED
@@ -4,6 +4,8 @@ from torch import cuda
4
  import streamlit as st
5
  from langchain_huggingface import HuggingFaceEmbeddings
6
  from appStore.prep_utils import get_client
 
 
7
 
8
  # get the device to be used eithe gpu or cpu
9
  device = 'cuda' if cuda.is_available() else 'cpu'
 
4
  import streamlit as st
5
  from langchain_huggingface import HuggingFaceEmbeddings
6
  from appStore.prep_utils import get_client
7
+ from qdrant_client import models # added
8
+
9
 
10
  # get the device to be used eithe gpu or cpu
11
  device = 'cuda' if cuda.is_available() else 'cpu'