Spaces:
Runtime error
Runtime error
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):
|