Update app.py

app.py

@@ -179,16 +179,9 @@ else:
 
     model_name = "Reyad-Ahmmed/hf-data-timeframe"
 
-    # Load the tokenizer from the subdirectory
-    tokenizer_save_path = AutoTokenizer.from_pretrained(model_name, subfolder="data-timeframe_tokenizer")
-    
-    # Load the model from the subdirectory
-    model_save_path = AutoModel.from_pretrained(model_name, subfolder="data-timeframe_model")
-
-
     # RobertaTokenizer.from_pretrained(model_save_path)
-    model = AutoModelForSequenceClassification.from_pretrained(model_save_path).to('cpu')
-    tokenizer = AutoTokenizer.from_pretrained(tokenizer_save_path)
+    model = AutoModelForSequenceClassification.from_pretrained(model_name, subfolder="data-timeframe_model").to('cpu')
+    tokenizer = AutoTokenizer.from_pretrained(model_name, subfolder="data-timeframe_tokenizer")
 
 #Define the label mappings (this must match the mapping used during training)
 label_mapping = {