StevenChen16 commited on
Commit
1083a3c
·
1 Parent(s): 52b36bc

add spaces.GPU(duration=120)

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -89,7 +89,7 @@ class RAGChatbot:
89
  [rest of your existing background prompt]
90
  '''
91
 
92
- @spaces.GPU
93
  def init_models(self):
94
  """Initialize the LLM and embedding models"""
95
  print("Initializing models...")
 
89
  [rest of your existing background prompt]
90
  '''
91
 
92
+ @spaces.GPU(duration=120)
93
  def init_models(self):
94
  """Initialize the LLM and embedding models"""
95
  print("Initializing models...")