Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|
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(
|