saibhossain commited on
Commit
be46249
·
1 Parent(s): 73523bb

update shear = true

Browse files
Files changed (1) hide show
  1. app.py +2 -6
app.py CHANGED
@@ -67,12 +67,8 @@ if PROPHET_AVAILABLE and os.path.exists(PROPHET_PATH):
67
  prophet_model = None
68
 
69
  if TENSORFLOW_AVAILABLE and os.path.exists(LSTM_PATH):
70
- print("✅ TensorFlow imported successfully")
71
  try:
72
- model = load_model("lstm_model.h5", compile=False)
73
- model.save("lstm_savedmodel", save_format="tf")
74
- lstm_model = load_model("models/lstm_savedmodel", compile=False)
75
- print("✅ LSTM model loaded:", LSTM_PATH)
76
  except Exception:
77
  lstm_model = None
78
 
@@ -336,4 +332,4 @@ with gr.Blocks(title="DataSynthis_ML_JobTask") as demo:
336
  back_table = gr.Dataframe(headers=["Model", "RMSE", "MAPE"])
337
  back_btn.click(fn=do_backtest, inputs=[test_days, arima_b, prophet_b, lstm_b], outputs=[back_plot, back_table])
338
 
339
- demo.launch()
 
67
  prophet_model = None
68
 
69
  if TENSORFLOW_AVAILABLE and os.path.exists(LSTM_PATH):
 
70
  try:
71
+ lstm_model = load_model(LSTM_PATH, compile=False)
 
 
 
72
  except Exception:
73
  lstm_model = None
74
 
 
332
  back_table = gr.Dataframe(headers=["Model", "RMSE", "MAPE"])
333
  back_btn.click(fn=do_backtest, inputs=[test_days, arima_b, prophet_b, lstm_b], outputs=[back_plot, back_table])
334
 
335
+ demo.launch(share=True)