add model testing
Browse files- server/model.py +8 -1
server/model.py
CHANGED
@@ -34,4 +34,11 @@ def translate(text: str):
|
|
34 |
sampling_temperature= 0,
|
35 |
)
|
36 |
|
37 |
-
return [''.join( detokenizeBatch(result.hypotheses[0]) ) for result in translated]
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
34 |
sampling_temperature= 0,
|
35 |
)
|
36 |
|
37 |
+
return [''.join( detokenizeBatch(result.hypotheses[0]) ) for result in translated]
|
38 |
+
|
39 |
+
|
40 |
+
if __name__ == "__main__":
|
41 |
+
translated = translate("ダンガンロンパ 希望の学園と絶望の高校生")
|
42 |
+
print(translated)
|
43 |
+
|
44 |
+
|