Update genra_incremental.py

genra_incremental.py

@@ -28,7 +28,7 @@ class LLMGenerator:
         self.tokenizer = tokenizer
         self.llm_name = llm_name
 
-    @spaces.GPU(duration=120)
+    @spaces.GPU(duration=1000)
     def generate_answer(self, texts, query, mode='validate'):
         template_texts =""
         for i, text in enumerate(texts):
@@ -64,7 +64,7 @@ class LLMGenerator:
 
 
 class QAIndexer:
-    @spaces.GPU(duration=60)
+    @spaces.GPU(duration=1000)
     def __init__(self, index_type, emb_model):
         self.document_embedder = SentenceTransformersDocumentEmbedder(model=emb_model)
         self.document_embedder.warm_up()
@@ -129,7 +129,7 @@ def rank_aggregation(aggregator, lists, k):
 
 
 class GenraPipeline:
-    @spaces.GPU(duration=60)
+    @spaces.GPU(duration=1000)
     def __init__(self, llm_name, emb_model, aggregator, contexts):
         self.qa_indexer = QAIndexer('in_memory', emb_model)
         self.qa_retriever = QARetriever(self.qa_indexer.document_store)