Update app.py

app.py

@@ -49,7 +49,7 @@ def check_existing_model(stock_symbol, start_date, end_date):
     state = repo_id in [model.modelId for model in hf_api.list_models()]
     return state
     
-@spaces.GPU(duration=300)
+@spaces.GPU(duration=120)
 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):
     repo_id = f"Q-bert/StockLlama-tuned{stock_symbol}-{start_date}_{end_date}"
 
@@ -124,7 +124,7 @@ def train_stock_model(stock_symbol, start_date, end_date, feature_range=(10, 100
     )
     return f"Training completed and model saved for {stock_symbol} from {start_date} to {end_date}."
 
-@spaces.GPU(duration=300)
+@spaces.GPU(duration=120)
 def gradio_train_stock_model(stock_symbol, start_date, end_date, feature_range_min, feature_range_max, data_seq_length, epochs, batch_size, learning_rate):
     feature_range = (feature_range_min, feature_range_max)
     result = train_stock_model(