Update back.py

back.py

@@ -42,14 +42,12 @@ label_mapping = {
 }
 
 
-#@st.cache_resource
 def load_setup(path_to_model: str, path_to_vocab: str) -> tuple[ModelType, TokenizerType]:
     loaded_model = torch.load(path_to_model, map_location=device)
     loaded_tokenizer = DistilBertTokenizer(path_to_vocab)
     return loaded_model, loaded_tokenizer
 
 
-#@st.cache_data
 def predict(model: ModelType, tokenizer: TokenizerType, input_text: str, max_length: int = 512) -> str:
     inputs = tokenizer.encode_plus(
         input_text,