Fix app

app.py

@@ -87,7 +87,7 @@ def date_format(dataframe):
 def group_to_three(dataframe):
   dataframe['Date'] = pd.to_datetime(dataframe['Date'])
   dataframe = dataframe.groupby([pd.Grouper(key='Date', freq='3D')])['Sales'].mean().round(2)
-  dataframe = dataframe.replace(0, pd.np.nan).dropna()
+  dataframe = dataframe.replace(0, np.nan).dropna()
 
   return dataframe
 
@@ -157,14 +157,14 @@ def sales_growth(dataframe, fittedValues):
 
 # TAPAS Model
 model_name = "google/tapas-large-finetuned-wtq"
+tokenizer = TapasTokenizer.from_pretrained(model_name)
+model = TapasForQuestionAnswering.from_pretrained(model_name, local_files_only=False)
 
-def load_tapas_model(model_name):
-  tokenizer = TapasTokenizer.from_pretrained(model_name)
-  model = TapasForQuestionAnswering.from_pretrained(model_name, local_files_only=False)
+def load_tapas_model(model, tokenizer):
   pipe = pipeline("table-question-answering", model=model, tokenizer=tokenizer)
   return pipe
 
-pipe = load_tapas_model(model_name)
+pipe = load_tapas_model(model, tokenizer)
 
 def get_answer(table, query):
     answers = pipe(table=table, query=query)