Emaad commited on
Commit
95c9069
1 Parent(s): 4877844

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -90,8 +90,8 @@ class model:
90
 
91
  formatted_predicted_sequence = formatted_predicted_sequence[0]
92
  formatted_predicted_sequence = formatted_predicted_sequence.replace("<pad>","")
93
- formatted_predicted_sequence = formatted_predicted_sequence.replace("<start>","")
94
- formatted_predicted_sequence = formatted_predicted_sequence.replace("<end>","")
95
 
96
  print(sequence_input)
97
  print(formatted_predicted_sequence)
 
90
 
91
  formatted_predicted_sequence = formatted_predicted_sequence[0]
92
  formatted_predicted_sequence = formatted_predicted_sequence.replace("<pad>","")
93
+ formatted_predicted_sequence = formatted_predicted_sequence.replace("<cls>","")
94
+ formatted_predicted_sequence = formatted_predicted_sequence.replace("<eos>","")
95
 
96
  print(sequence_input)
97
  print(formatted_predicted_sequence)