MFBDA commited on
Commit
60de325
verified
1 Parent(s): c6bb5e0

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +30 -0
app.py ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import TimeSeriesTransformerForPrediction, TimeSeriesTransformerConfig
3
+ 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(",")]
12
+
13
+ # Preparar dados (ajuste conforme o formato esperado pelo modelo)
14
+ data = pd.Series(historico)
15
+
16
+ # Gerar previs茫o (exemplo simplificado)
17
+ forecast = model.predict(data, prediction_length=3) # Previs茫o para 3 meses
18
+ return np.round(forecast.mean, 2).tolist()
19
+
20
+ # Interface Gradio
21
+ iface = gr.Interface(
22
+ fn=prever_vendas,
23
+ inputs=gr.Textbox(label="Hist贸rico de Vendas (separado por v铆rgulas)"),
24
+ outputs=gr.Textbox(label="Previs茫o para os Pr贸ximos 3 Meses"),
25
+ examples=[
26
+ ["140,155,160,145,150,165,170,160,175,160,155,170"], # Exemplo do Produto C
27
+ ]
28
+ )
29
+
30
+ iface.launch()