Penality commited on
Commit
24060f0
·
verified ·
1 Parent(s): 4df7749

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -16,6 +16,9 @@ load_dotenv()
16
 
17
  API_URL = f"https://b4e0-105-160-26-239.ngrok-free.app/post_embeddings"
18
 
 
 
 
19
  # Set up Together.AI API Key (Replace with your actual key)
20
  assert os.getenv("TOGETHER_API_KEY"), "api key missing"
21
 
@@ -48,7 +51,8 @@ def store_document_data(PDF_FILE):
48
  print(f"sending to {API_URL}")
49
 
50
  try:
51
- response = requests.post(API_URL, json={"index": embedding})
 
52
  response_data = response.json()
53
  return response_data
54
  except requests.exceptions.RequestException as e:
 
16
 
17
  API_URL = f"https://b4e0-105-160-26-239.ngrok-free.app/post_embeddings"
18
 
19
+ # FAISS index setup
20
+ DIM = 768 # Adjust based on the embedding model
21
+
22
  # Set up Together.AI API Key (Replace with your actual key)
23
  assert os.getenv("TOGETHER_API_KEY"), "api key missing"
24
 
 
51
  print(f"sending to {API_URL}")
52
 
53
  try:
54
+ index = faiss.IndexFlatL2(embedding.shape[1])
55
+ response = requests.post(API_URL, json={"index": index})
56
  response_data = response.json()
57
  return response_data
58
  except requests.exceptions.RequestException as e: