sugiv commited on
Commit
a34fd06
·
1 Parent(s): 2fa014f

Adding a simple monkey search for Leetcode - Darn LeetMonkey

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -7,15 +7,16 @@ from sentence_transformers import SentenceTransformer
7
  import transformers
8
  transformers.logging.set_verbosity_error()
9
 
10
- # Initialize Pinecone
11
- PINECONE_API_KEY = "your_pinecone_api_key"
 
12
  pc = Pinecone(api_key=PINECONE_API_KEY)
13
 
14
  index_name = "leetmonkey-sparse-dense"
15
  index = pc.Index(index_name)
16
 
17
  # Initialize models
18
- device = 'cuda' if torch.cuda.is_available() else 'cpu'
19
  splade = SpladeEncoder(device=device)
20
  dense_model = SentenceTransformer('sentence-transformers/all-Mpnet-base-v2', device=device)
21
 
@@ -91,4 +92,4 @@ iface = gr.Interface(
91
  )
92
 
93
  # Launch the app
94
- iface.launch()
 
7
  import transformers
8
  transformers.logging.set_verbosity_error()
9
 
10
+ import os
11
+ PINECONE_API_KEY = os.environ.get('PINECONE_API_KEY')
12
+
13
  pc = Pinecone(api_key=PINECONE_API_KEY)
14
 
15
  index_name = "leetmonkey-sparse-dense"
16
  index = pc.Index(index_name)
17
 
18
  # Initialize models
19
+ device = 'cpu'
20
  splade = SpladeEncoder(device=device)
21
  dense_model = SentenceTransformer('sentence-transformers/all-Mpnet-base-v2', device=device)
22
 
 
92
  )
93
 
94
  # Launch the app
95
+ iface.launch(share=True)