developersajidbashir commited on
Commit
899523b
·
verified ·
1 Parent(s): cde78d6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -7
app.py CHANGED
@@ -15,7 +15,6 @@ import os
15
  cred = credentials.Certificate("credentials.json")
16
  firebase_admin.initialize_app(cred, {"databaseURL": "https://socail-swap-default-rtdb.asia-southeast1.firebasedatabase.app/"})
17
  ref = db.reference()
18
- stopmodel = False
19
  buy_signals = []
20
  sell_signals = []
21
 
@@ -139,9 +138,5 @@ if _name_ == "_main_":
139
 
140
  env = DummyVecEnv([lambda: TradingEnv(data)])
141
 
142
- if os.path.exists("./ppo_trading_agent.zip"):
143
- model = PPO.load("ppo_trading_agent", env=env)
144
- else:
145
- model = PPO('MlpPolicy', env, verbose=1)
146
- model.learn(total_timesteps=10000)
147
-     run_model()
 
15
  cred = credentials.Certificate("credentials.json")
16
  firebase_admin.initialize_app(cred, {"databaseURL": "https://socail-swap-default-rtdb.asia-southeast1.firebasedatabase.app/"})
17
  ref = db.reference()
 
18
  buy_signals = []
19
  sell_signals = []
20
 
 
138
 
139
  env = DummyVecEnv([lambda: TradingEnv(data)])
140
 
141
+ model = PPO.load("ppo_trading_agent", env=env)
142
+ run_model()