Update app.py

app.py

@@ -101,7 +101,7 @@ def embedding_shift(original_embedding,shift_embeddings,prefix_embedding,suffix_
         )
     return input_embeddings
 
-@spaces.GPU(duration=2)
+@spaces.GPU(duration=10)
 def engine(input_embeds):
     m.to("cuda")
     output_text = []
@@ -120,7 +120,7 @@ def engine(input_embeds):
             output_text += tok.batch_decode(outputs, skip_special_tokens=True)
     return output_text
     
-@spaces.GPU(duration=2)
+@spaces.GPU(duration=10)
 def chat_engine(input_ids):
     m.to("cuda")
     prompt_length=len(input_ids[0])