Ankerkraut commited on
Commit
fd26237
·
1 Parent(s): 4cdc9ad

increase duration

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -40,7 +40,7 @@ client.add(collection_name="recipes",
40
  model_name = "LeoLM/leo-hessianai-70b-chat"
41
 
42
  last_messages = []
43
- @spaces.GPU
44
  def load_model():
45
  ankerbot_model = AutoModelForCausalLM.from_pretrained(
46
  model_name,
@@ -193,7 +193,7 @@ def interactive_chat(query):
193
  print(f"last messages: {last_messages}")
194
  print()
195
  return answer
196
- @spaces.GPU
197
  def get_answers():
198
  answers = []
199
  last_messages = []
 
40
  model_name = "LeoLM/leo-hessianai-70b-chat"
41
 
42
  last_messages = []
43
+ @spaces.GPU(duration=120)
44
  def load_model():
45
  ankerbot_model = AutoModelForCausalLM.from_pretrained(
46
  model_name,
 
193
  print(f"last messages: {last_messages}")
194
  print()
195
  return answer
196
+ @spaces.GPU(duration=1200)
197
  def get_answers():
198
  answers = []
199
  last_messages = []