Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -70,6 +70,7 @@ class DocumentRetrievalAndGeneration:
|
|
| 70 |
generated_ids = future.result(timeout=60) # Timeout set to 60 seconds
|
| 71 |
return generated_ids
|
| 72 |
except concurrent.futures.TimeoutError:
|
|
|
|
| 73 |
raise TimeoutError("Text generation process timed out")
|
| 74 |
|
| 75 |
def query_and_generate_response(self, query):
|
|
|
|
| 70 |
generated_ids = future.result(timeout=60) # Timeout set to 60 seconds
|
| 71 |
return generated_ids
|
| 72 |
except concurrent.futures.TimeoutError:
|
| 73 |
+
return "Text generation process timed out"
|
| 74 |
raise TimeoutError("Text generation process timed out")
|
| 75 |
|
| 76 |
def query_and_generate_response(self, query):
|