Spaces:

Circhastic
/

sales-forecasting

Running

Circhastic commited on Dec 1, 2023

Commit

506e25b

1 Parent(s): 1898dec

Fix app

Files changed (1) hide show

app.py CHANGED Viewed

@@ -160,7 +160,6 @@ 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
 def load_tapas_model(model, tokenizer):
   pipe = pipeline("table-question-answering", model=model, tokenizer=tokenizer)
   return pipe

 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