Update app.py

app.py

@@ -10,7 +10,7 @@ from huggingface_hub import hf_hub_download
 # 1. LOAD MODEL, TOKENIZER, AND LABEL ENCODER
 # =============================================================================
 # Define the path to your model repository
-model_path = "Tarive/esm2_t12_35M_UR50D-5k-families-balanced-augmented-weighted_optimized"
+model_path = "Tarive/esm2_t12_35M_UR50D-finetuned-pfam-1k"
 
 print("Loading tokenizer...")
 tokenizer = AutoTokenizer.from_pretrained(model_path)
@@ -24,7 +24,7 @@ print(f"Model loaded on device: {device}")
 
 # Download and load the label encoder
 print("Downloading and loading label encoder...")
-encoder_path = hf_hub_download(repo_id=model_path, filename="label_encoder_5k-2.pkl")
+encoder_path = hf_hub_download(repo_id=model_path, filename="label_encoder.pkl")
 with open(encoder_path, "rb") as f:
     label_encoder = pickle.load(f)
 print("Label encoder loaded.")