Ron Au commited on
Commit
ff8ee9b
1 Parent(s): 9ccba4b

feat(prompts): Use original Russian labels

Browse files
Files changed (1) hide show
  1. modules/inference.py +11 -11
modules/inference.py CHANGED
@@ -33,17 +33,17 @@ print(f'{strftime("%Y-%m-%d %H:%M:%S", gmtime())} Ready for inference')
33
 
34
  def english_to_russian(english_string):
35
  word_map = {
36
- "grass": "трава",
37
- "fire": "Пожар",
38
- "water": "вода",
39
- "lightning": "молния",
40
- "fighting": "борьба",
41
- "psychic": "психический",
42
- "colorless": "бесцветный",
43
- "darkness": "темнота",
44
- "metal": "металл",
45
- "dragon": "Дракон",
46
- "fairy": "сказочный"
47
  }
48
 
49
  return word_map[english_string.lower()]
 
33
 
34
  def english_to_russian(english_string):
35
  word_map = {
36
+ "grass": "Покемон трава",
37
+ "fire": "Покемон огня",
38
+ "water": "Покемон в воду",
39
+ "lightning": "Покемон электрического типа",
40
+ "fighting": "Покемон боевого типа",
41
+ "psychic": "Покемон психического типа",
42
+ "colorless": "Покемон нормального типа",
43
+ "darkness": "Покемон темного типа",
44
+ "metal": "Покемон из стали типа",
45
+ "dragon": "Покемон типа дракона",
46
+ "fairy": "Покемон фея"
47
  }
48
 
49
  return word_map[english_string.lower()]