Spaces:
Running
Running
Ron Au
commited on
Commit
•
ff8ee9b
1
Parent(s):
9ccba4b
feat(prompts): Use original Russian labels
Browse files- 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()]
|