playmak3r commited on
Commit
96aee55
·
1 Parent(s): 0918613

add model testing

Browse files
Files changed (1) hide show
  1. 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
+