Guardar mis cambios locales

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import pandas as pd
 import numpy as np
 from datetime import datetime
+import joblib
 import pmdarima as pm
 from pmdarima import auto_arima
 import plotly.graph_objects as go
@@ -9,7 +10,7 @@ from transformers import pipeline, TapasTokenizer, TapasForQuestionAnswering
 
 # Load the TAPAS Model
 def load_tapas_model():
-    model_name = "google/tapas-large-finetuned-wtq"
+    model_name = joblib.load('arima_sales_model.pkl')
     tokenizer = TapasTokenizer.from_pretrained(model_name)
     model = TapasForQuestionAnswering.from_pretrained(model_name)
     pipe = pipeline("table-question-answering", model=model, tokenizer=tokenizer)