MFBDA commited on
Commit
a529bcd
·
verified ·
1 Parent(s): 4571224

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,14 +1,13 @@
1
  import gradio as gr
2
  from transformers import TimeSeriesTransformerForPrediction, TimeSeriesTransformerConfig
3
  import torch
4
- import pandas as pd
5
  import numpy as np
6
 
7
  # Carregar configuração
8
  config = TimeSeriesTransformerConfig.from_pretrained("google/timesfm-2.0-500m-pytorch")
9
  config.prediction_length = 3
10
  config.context_length = 12
11
- config.lags_sequence = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12] # Ajuste os lags para serem menores ou iguais ao context_length
12
 
13
  # Carregar modelo com a configuração ajustada
14
  model = TimeSeriesTransformerForPrediction.from_pretrained(
 
1
  import gradio as gr
2
  from transformers import TimeSeriesTransformerForPrediction, TimeSeriesTransformerConfig
3
  import torch
 
4
  import numpy as np
5
 
6
  # Carregar configuração
7
  config = TimeSeriesTransformerConfig.from_pretrained("google/timesfm-2.0-500m-pytorch")
8
  config.prediction_length = 3
9
  config.context_length = 12
10
+ config.lags_sequence = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11] # Lags menores que context_length
11
 
12
  # Carregar modelo com a configuração ajustada
13
  model = TimeSeriesTransformerForPrediction.from_pretrained(