MFBDA commited on
Commit
9627e09
·
verified ·
1 Parent(s): 1afd6ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -4,8 +4,10 @@ import pandas as pd
4
  import numpy as np
5
 
6
  # Carregar o modelo
7
- model = TimeSeriesTransformerForPrediction.from_pretrained("google/timesfm-2.0-500m-pytorch")
8
-
 
 
9
  def prever_vendas(historico):
10
  # Converter entrada em lista de números
11
  historico = [float(x) for x in historico.split(",")]
 
4
  import numpy as np
5
 
6
  # Carregar o modelo
7
+ model = TimeSeriesTransformerForPrediction.from_pretrained(
8
+ "google/timesfm-2.0-500m-pytorch",
9
+ torch_dtype="auto" # Garante compatibilidade
10
+ )
11
  def prever_vendas(historico):
12
  # Converter entrada em lista de números
13
  historico = [float(x) for x in historico.split(",")]