Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -77,9 +77,29 @@ def retrieve_document(query):
|
|
| 77 |
embeddings_ = requests.get(API_URL_EMBEDDINGS)
|
| 78 |
metadata_ = requests.get(API_URL_METADATA)
|
| 79 |
|
| 80 |
-
|
| 81 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 82 |
print(embeddings_file, metadata_file)
|
|
|
|
| 83 |
# Generate query embedding
|
| 84 |
query_embedding = embedding_model.encode([query]).astype(np.float32)
|
| 85 |
|
|
|
|
| 77 |
embeddings_ = requests.get(API_URL_EMBEDDINGS)
|
| 78 |
metadata_ = requests.get(API_URL_METADATA)
|
| 79 |
|
| 80 |
+
# Check for errors before parsing JSON
|
| 81 |
+
if embeddings_.status_code != 200:
|
| 82 |
+
print(f"Error fetching embeddings: {embeddings_.status_code} - {embeddings_.text}")
|
| 83 |
+
return None
|
| 84 |
+
|
| 85 |
+
if metadata_.status_code != 200:
|
| 86 |
+
print(f"Error fetching metadata: {metadata_.status_code} - {metadata_.text}")
|
| 87 |
+
return None
|
| 88 |
+
|
| 89 |
+
try:
|
| 90 |
+
embeddings_file = embeddings_.json()
|
| 91 |
+
except requests.exceptions.JSONDecodeError as e:
|
| 92 |
+
print(f"Error decoding embeddings JSON: {e}")
|
| 93 |
+
return None
|
| 94 |
+
|
| 95 |
+
try:
|
| 96 |
+
metadata_file = metadata_.json()
|
| 97 |
+
except requests.exceptions.JSONDecodeError as e:
|
| 98 |
+
print(f"Error decoding metadata JSON: {e}")
|
| 99 |
+
return None
|
| 100 |
+
|
| 101 |
print(embeddings_file, metadata_file)
|
| 102 |
+
|
| 103 |
# Generate query embedding
|
| 104 |
query_embedding = embedding_model.encode([query]).astype(np.float32)
|
| 105 |
|