Update app.py

app.py

@@ -188,11 +188,15 @@ def find_example_files():
 #  Load model from HF Hub
 # ==========================
 
-MODEL_ID = "OSS-Forge/codet5p-770m-pyresbugs" 
+BASE_MODEL_ID = "Salesforce/codet5p-770m"            
+FINETUNED_MODEL_ID = "OSS-Forge/codet5p-770m-pyresbugs"  
+
+print(f"Loading tokenizer from base model: {BASE_MODEL_ID}")
+tokenizer = AutoTokenizer.from_pretrained(BASE_MODEL_ID)
+
+print(f"Loading fine-tuned model weights from: {FINETUNED_MODEL_ID}")
+model = AutoModelForSeq2SeqLM.from_pretrained(FINETUNED_MODEL_ID)
 
-print(f"Loading tokenizer and model from: {MODEL_ID}")
-tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
-model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_ID)
 
 
 # ==========================