bintangyosua commited on
Commit
fd7dca3
·
verified ·
1 Parent(s): 71f46e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = "Progressive" if predicted_class == 1 else "Conservative"
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