FractalAIR commited on
Commit
addf70e
·
verified ·
1 Parent(s): ae84060

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ except Exception as e:
32
  # Ensure model is on GPU
33
  #model = model.to(device)
34
 
35
- @spaces.GPU(duration=180)
36
  def generate_response(user_message, max_tokens, temperature, top_k, top_p, repetition_penalty, history_state):
37
  if not user_message.strip():
38
  logger.info("Empty message received, returning history unchanged.")
 
32
  # Ensure model is on GPU
33
  #model = model.to(device)
34
 
35
+ @spaces.GPU(duration=300)
36
  def generate_response(user_message, max_tokens, temperature, top_k, top_p, repetition_penalty, history_state):
37
  if not user_message.strip():
38
  logger.info("Empty message received, returning history unchanged.")