Hugging Face's logo

Spaces:
bkhmsi
/
Partial-Arabic-Diacritization
Running

App Files Community
bkhmsi commited on
Commit
cd87bdb
•
1 Parent(s): d800c0d

bugfix

Browse files
Files changed (1) hide show
  1. predict.py +1 -1
predict.py CHANGED
@@ -80,7 +80,7 @@ class PredictTri(Predictor):
80
 
81
  def predict_majority_vote(self):
82
  y_gen_diac, y_gen_tanween, y_gen_shadda = self.model.predict(self.data_loader)
83
- diacritized_lines = self.coalesce_votes_by_majority(y_gen_diac, y_gen_tanween, y_gen_shadda)
84
  return diacritized_lines
85
 
86
  def predict_majority_vote_context_contrastive(self, overwrite_cache=False):
 
80
 
81
  def predict_majority_vote(self):
82
  y_gen_diac, y_gen_tanween, y_gen_shadda = self.model.predict(self.data_loader)
83
+ diacritized_lines, _ = self.coalesce_votes_by_majority(y_gen_diac, y_gen_tanween, y_gen_shadda)
84
  return diacritized_lines
85
 
86
  def predict_majority_vote_context_contrastive(self, overwrite_cache=False):