Spaces:
Sleeping
Sleeping
bintangyosua
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -656,7 +656,7 @@ def __(embedding_models, get_doc_embedding, loaded_model, preprocess_text):
|
|
656 |
embedded = embedded.reshape(1, 1, -1)
|
657 |
prediction = loaded_model.predict(embedded)
|
658 |
predicted_class = prediction.argmax(axis=-1)
|
659 |
-
predicted_class = "
|
660 |
return predicted_class
|
661 |
return (try_predict,)
|
662 |
|
|
|
656 |
embedded = embedded.reshape(1, 1, -1)
|
657 |
prediction = loaded_model.predict(embedded)
|
658 |
predicted_class = prediction.argmax(axis=-1)
|
659 |
+
predicted_class = "Liberal" if predicted_class == 1 else "Conservative"
|
660 |
return predicted_class
|
661 |
return (try_predict,)
|
662 |
|