Bugfix

app.py

@@ -42,7 +42,7 @@ def _predict(text, model, vectorizer):
     X_test = vectorizer.transform([clean_text])
     X_test = vectorizer.transform([clean_text])
 
 
     # Predict outputs
     # Predict outputs
-    labels = [f"PR{i}" for i in range(7)]
+    labels = [f"BR{i}" for i in range(7)]
     probs = model.predict_proba(X_test)
     probs = model.predict_proba(X_test)
     probs = [prob[0][1] for prob in probs]
     probs = [prob[0][1] for prob in probs]
     probs = [
     probs = [