Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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):
|