CAGmllab commited on
Commit
fe57d8f
1 Parent(s): d4456cd

Update styleformer/styleformer.py

Browse files
Files changed (1) hide show
  1. styleformer/styleformer.py +1 -1
styleformer/styleformer.py CHANGED
@@ -40,7 +40,7 @@ class Styleformer():
40
  else:
41
  print("Only CTF, FTC, ATP and PTA are supported in the pre-release...stay tuned")
42
 
43
- def transfer(self, input_sentence, quality_filter=0.95, max_candidates=1:
44
  if self.model_loaded:
45
  device = "cpu"
46
 
 
40
  else:
41
  print("Only CTF, FTC, ATP and PTA are supported in the pre-release...stay tuned")
42
 
43
+ def transfer(self, input_sentence, quality_filter=0.95, max_candidates=1):
44
  if self.model_loaded:
45
  device = "cpu"
46