pendar02 commited on
Commit
d1c4e7d
·
verified ·
1 Parent(s): e9d2a43

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -13
app.py CHANGED
@@ -85,19 +85,6 @@ def load_model(model_type):
85
 
86
  model.eval()
87
  return model, tokenizer
88
-
89
- except Exception as e:
90
- # Fallback to base model if PEFT loading fails
91
- st.warning(f"Error loading PEFT model: {str(e)}. Falling back to base model.")
92
- if model_type == "summarize":
93
- model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn")
94
- tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-cnn")
95
- else:
96
- model = AutoModelForSeq2SeqLM.from_pretrained("GanjinZero/biobart-base")
97
- tokenizer = AutoTokenizer.from_pretrained("GanjinZero/biobart-base")
98
-
99
- model.eval()
100
- return model, tokenizer
101
 
102
  @st.cache_data
103
  def process_excel(uploaded_file):
 
85
 
86
  model.eval()
87
  return model, tokenizer
 
 
 
 
 
 
 
 
 
 
 
 
 
88
 
89
  @st.cache_data
90
  def process_excel(uploaded_file):