Spaces:
Sleeping
Sleeping
Update model.py
Browse files
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("
|
| 18 |
except Exception as e:
|
| 19 |
-
print(f"
|
| 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"
|
| 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}
|