Spaces:
Sleeping
Sleeping
WilliamGazeley
commited on
Commit
·
cec1525
1
Parent(s):
eac07bb
Add max retry
Browse files- src/app.py +3 -2
src/app.py
CHANGED
@@ -107,7 +107,7 @@ def main():
|
|
107 |
|
108 |
input_text = st.text_area("Enter your text here:", value="", height=200)
|
109 |
|
110 |
-
def get_response(input_text: str) -> str:
|
111 |
try:
|
112 |
for output in flow.stream({"question": input_text}):
|
113 |
for key, value in output.items():
|
@@ -117,7 +117,8 @@ def main():
|
|
117 |
except Exception as e:
|
118 |
logger.error(e)
|
119 |
logger.info("Retrying..")
|
120 |
-
|
|
|
121 |
|
122 |
if st.button("Generate"):
|
123 |
if input_text:
|
|
|
107 |
|
108 |
input_text = st.text_area("Enter your text here:", value="", height=200)
|
109 |
|
110 |
+
def get_response(input_text: str, depth: int = 1) -> str:
|
111 |
try:
|
112 |
for output in flow.stream({"question": input_text}):
|
113 |
for key, value in output.items():
|
|
|
117 |
except Exception as e:
|
118 |
logger.error(e)
|
119 |
logger.info("Retrying..")
|
120 |
+
if depth < 5:
|
121 |
+
return get_response(input_text, depth + 1)
|
122 |
|
123 |
if st.button("Generate"):
|
124 |
if input_text:
|