庞喆 commited on
Commit
a00a030
·
1 Parent(s): ac429e6

load env variables

Browse files
Files changed (1) hide show
  1. search.py +3 -2
search.py CHANGED
@@ -6,6 +6,7 @@ import io
6
  import gradio as gr
7
  from pinecone import Pinecone
8
  import logging
 
9
 
10
  # Setup basic configuration for logging
11
  logging.basicConfig(filename='embedding_errors.log', level=logging.ERROR,
@@ -16,8 +17,8 @@ DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
16
 
17
  # Initialize CLIP model and Pinecone
18
  model, preprocess = clip.load("ViT-B/32", device=DEVICE)
19
- pinecone = Pinecone(api_key=PINECONE_API_KEY)
20
- index = pinecone.Index(PINECONE_INDEX_NAME)
21
 
22
  def generate_combined_embeddings(image_url, text_description):
23
  """ Generate combined embeddings for image and text using CLIP. """
 
6
  import gradio as gr
7
  from pinecone import Pinecone
8
  import logging
9
+ import os
10
 
11
  # Setup basic configuration for logging
12
  logging.basicConfig(filename='embedding_errors.log', level=logging.ERROR,
 
17
 
18
  # Initialize CLIP model and Pinecone
19
  model, preprocess = clip.load("ViT-B/32", device=DEVICE)
20
+ pinecone = Pinecone(api_key=os.getenv('PINECONE_API_KEY'))
21
+ index = pinecone.Index(os.getenv('PINECONE_INDEX_NAME'))
22
 
23
  def generate_combined_embeddings(image_url, text_description):
24
  """ Generate combined embeddings for image and text using CLIP. """