Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update appStore/rag_utils.py
Browse files- appStore/rag_utils.py +8 -0
appStore/rag_utils.py
CHANGED
@@ -124,5 +124,13 @@ def get_rag_answer(query, top_results, endpoint, token):
|
|
124 |
if "Answer:" in answer:
|
125 |
answer = answer.split("Answer:")[-1].strip()
|
126 |
return answer
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
127 |
else:
|
128 |
return f"Error in generating answer: {response.text}"
|
|
|
124 |
if "Answer:" in answer:
|
125 |
answer = answer.split("Answer:")[-1].strip()
|
126 |
return answer
|
127 |
+
elif response.status_code == 503:
|
128 |
+
# Custom message with a larger llama icon and red highlighted text
|
129 |
+
return (
|
130 |
+
"<span style='color: red;'>"
|
131 |
+
"<span style='font-size: 2em;'>🦙</span> tzzz tzzz I'm currently sleeping. "
|
132 |
+
"Please come back in 10 minutes, and I'll be fully awake to answer your question."
|
133 |
+
"</span>"
|
134 |
+
)
|
135 |
else:
|
136 |
return f"Error in generating answer: {response.text}"
|