Update app.py

app.py

@@ -188,14 +188,18 @@ def find_example_files():
 #  Load model from HF Hub
 # ==========================
 
-BASE_MODEL_ID = "Salesforce/codet5p-770m"            
-FINETUNED_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)
+model = AutoModelForSeq2SeqLM.from_pretrained(
+    FINETUNED_MODEL_ID,
+    ignore_mismatched_sizes=True,   
+)
+