Q-bert commited on
Commit
d3f2a71
1 Parent(s): 8fe3995

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ def check_existing_model(stock_symbol, start_date, end_date):
49
  state = repo_id in [model.modelId for model in hf_api.list_models()]
50
  return state
51
 
52
- @spaces.GPU
53
  def train_stock_model(stock_symbol, start_date, end_date, feature_range=(10, 100), data_seq_length=256, epochs=10, batch_size=16, learning_rate=2e-4):
54
  repo_id = f"Q-bert/StockLlama-tuned-{stock_symbol}-{stock_symbol}-{start_date}_{end_date}"
55
 
 
49
  state = repo_id in [model.modelId for model in hf_api.list_models()]
50
  return state
51
 
52
+ @spaces.GPU(duration=300)
53
  def train_stock_model(stock_symbol, start_date, end_date, feature_range=(10, 100), data_seq_length=256, epochs=10, batch_size=16, learning_rate=2e-4):
54
  repo_id = f"Q-bert/StockLlama-tuned-{stock_symbol}-{stock_symbol}-{start_date}_{end_date}"
55