Spaces:
Sleeping
Sleeping
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 |
|