Update app.py

app.py

@@ -85,6 +85,19 @@ def load_model(model_type):
         
         model.eval()
         return model, tokenizer
+    
+    except Exception as e:
+        # Fallback to base model if PEFT loading fails
+        st.warning(f"Error loading PEFT model: {str(e)}. Falling back to base model.")
+        if model_type == "summarize":
+            model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn")
+            tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-cnn")
+        else:
+            model = AutoModelForSeq2SeqLM.from_pretrained("GanjinZero/biobart-base")
+            tokenizer = AutoTokenizer.from_pretrained("GanjinZero/biobart-base")
+        
+        model.eval()
+        return model, tokenizer
 
 @st.cache_data
 def process_excel(uploaded_file):