Spaces:

Ritesh-hf
/

rag-api

Running on Zero

Ritesh-hf commited on 21 days ago

Commit

8870bd5

•

1 Parent(s): 21901c4

update GPU duration

Files changed (1) hide show

app.py CHANGED Viewed

@@ -120,7 +120,7 @@ conversational_rag_chain = RunnableWithMessageHistory(
     output_messages_key="answer",
 )
-@spaces.GPU(duration=10)
 def handle_message(question, history={}):
     response = ''
     chain = conversational_rag_chain.pick("answer")

     output_messages_key="answer",
 )
+@spaces.GPU(duration=5)
 def handle_message(question, history={}):
     response = ''
     chain = conversational_rag_chain.pick("answer")