sree4411 commited on
Commit
adca082
·
verified ·
1 Parent(s): c5c9320

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -3
app.py CHANGED
@@ -4,9 +4,12 @@ import streamlit as st
4
  import pickle
5
 
6
  # Load models using pickle
7
- with open("tfidf_vectorizer(1).pkl", "rb") as f:
8
- vectorizer = pickle.load(f)
9
-
 
 
 
10
  with open("model(1).pkl", "rb") as f:
11
  model = pickle.load(f)
12
 
 
4
  import pickle
5
 
6
  # Load models using pickle
7
+ try:
8
+ with open("tfidf_vectorizer(1).pkl", "rb") as f:
9
+ vectorizer = pickle.load(f)
10
+ except FileNotFoundError:
11
+ st.error("❌ 'tfidf_vectorizer(1).pkl' not found. Please make sure the file is uploaded to the root directory.")
12
+ st.stop()
13
  with open("model(1).pkl", "rb") as f:
14
  model = pickle.load(f)
15