Fix app

app.py

@@ -170,12 +170,12 @@ def sales_growth(dataframe, fittedValues):
     return sales_growth
 
 # 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)
 
 @st.cache_resource
 def load_tapas_model(model, tokenizer):
+  model_name = "google/tapas-large-finetuned-wtq"
+  tokenizer = TapasTokenizer.from_pretrained(model_name)
+  model = TapasForQuestionAnswering.from_pretrained(model_name, local_files_only=False)
   pipe = pipeline("table-question-answering", model=model, tokenizer=tokenizer)
   return pipe