aiqtech commited on
Commit
9ff4e6c
·
verified ·
1 Parent(s): f05802f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -29,9 +29,12 @@ def predict_future_prices(ticker, periods=1825): # 5년간의 데이터 예측
29
  future = model.make_future_dataframe(periods=periods, freq='D')
30
  forecast = model.predict(future)
31
 
 
 
 
32
  # 예측 결과를 그래프로 표현
33
  fig = go.Figure(data=[go.Scatter(x=forecast['ds'], y=forecast['yhat'], mode='lines', name='Forecast')])
34
- return fig, forecast[['ds', 'yhat', 'yhat_lower', 'yhat_upper']] # Pydantic과 같은 엄격한 형식 검사 없이 DataFrame으로 직접 전달
35
 
36
  # Gradio 인터페이스 설정 및 실행
37
  with gr.Blocks() as app:
 
29
  future = model.make_future_dataframe(periods=periods, freq='D')
30
  forecast = model.predict(future)
31
 
32
+ # ds를 문자열로 변환
33
+ forecast['ds'] = forecast['ds'].dt.strftime('%Y-%m-%d')
34
+
35
  # 예측 결과를 그래프로 표현
36
  fig = go.Figure(data=[go.Scatter(x=forecast['ds'], y=forecast['yhat'], mode='lines', name='Forecast')])
37
+ return fig, forecast[['ds', 'yhat', 'yhat_lower', 'yhat_upper']]
38
 
39
  # Gradio 인터페이스 설정 및 실행
40
  with gr.Blocks() as app: