dev2004v commited on
Commit
c369fec
·
verified ·
1 Parent(s): 334b424

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +3 -3
model.py CHANGED
@@ -14,9 +14,9 @@ try:
14
  model = joblib.load(MODEL_PATH)
15
  vectorizer = joblib.load(VECTORIZER_PATH)
16
  svd = joblib.load(SVD_PATH)
17
- print(" Model, vectorizer, and SVD loaded successfully!")
18
  except Exception as e:
19
- print(f" Error loading model files: {e}")
20
  exit(1)
21
 
22
  def predict_text(text: str) -> dict:
@@ -31,5 +31,5 @@ def predict_text(text: str) -> dict:
31
 
32
  return {"generated": int(prediction), "probability": probability}
33
  except Exception as e:
34
- print(f" Prediction error: {e}")
35
  return {"generated": -1, "probability": 0.0}
 
14
  model = joblib.load(MODEL_PATH)
15
  vectorizer = joblib.load(VECTORIZER_PATH)
16
  svd = joblib.load(SVD_PATH)
17
+ print(" Model, vectorizer, and SVD loaded successfully!")
18
  except Exception as e:
19
+ print(f" Error loading model files: {e}")
20
  exit(1)
21
 
22
  def predict_text(text: str) -> dict:
 
31
 
32
  return {"generated": int(prediction), "probability": probability}
33
  except Exception as e:
34
+ print(f" Prediction error: {e}")
35
  return {"generated": -1, "probability": 0.0}