Update styleformer/styleformer.py
Browse files
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=
|
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 |
|