Spaces:
Sleeping
Sleeping
DenkingOfficial
commited on
Commit
·
202d636
1
Parent(s):
4bf9540
Updated code and formatted breeds
Browse files- cat_breeds_dict.py +317 -107
- main_app.py +14 -5
cat_breeds_dict.py
CHANGED
@@ -1,107 +1,317 @@
|
|
1 |
-
CAT_BREEDS = [
|
2 |
-
|
3 |
-
|
4 |
-
|
5 |
-
|
6 |
-
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
-
|
14 |
-
|
15 |
-
|
16 |
-
|
17 |
-
|
18 |
-
|
19 |
-
|
20 |
-
|
21 |
-
|
22 |
-
|
23 |
-
|
24 |
-
|
25 |
-
|
26 |
-
|
27 |
-
|
28 |
-
|
29 |
-
|
30 |
-
|
31 |
-
|
32 |
-
|
33 |
-
|
34 |
-
|
35 |
-
|
36 |
-
|
37 |
-
|
38 |
-
|
39 |
-
|
40 |
-
|
41 |
-
|
42 |
-
|
43 |
-
|
44 |
-
|
45 |
-
|
46 |
-
|
47 |
-
|
48 |
-
|
49 |
-
|
50 |
-
|
51 |
-
|
52 |
-
|
53 |
-
|
54 |
-
|
55 |
-
|
56 |
-
|
57 |
-
|
58 |
-
|
59 |
-
|
60 |
-
|
61 |
-
|
62 |
-
|
63 |
-
|
64 |
-
|
65 |
-
|
66 |
-
|
67 |
-
|
68 |
-
|
69 |
-
|
70 |
-
|
71 |
-
|
72 |
-
|
73 |
-
|
74 |
-
|
75 |
-
|
76 |
-
|
77 |
-
|
78 |
-
|
79 |
-
|
80 |
-
|
81 |
-
|
82 |
-
|
83 |
-
|
84 |
-
|
85 |
-
|
86 |
-
|
87 |
-
|
88 |
-
|
89 |
-
|
90 |
-
|
91 |
-
|
92 |
-
|
93 |
-
|
94 |
-
|
95 |
-
|
96 |
-
|
97 |
-
|
98 |
-
|
99 |
-
|
100 |
-
|
101 |
-
|
102 |
-
|
103 |
-
|
104 |
-
|
105 |
-
|
106 |
-
|
107 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
CAT_BREEDS = [
|
2 |
+
'Бенгальская', 'Бомбейская', 'Британская короткошерстная',
|
3 |
+
'Бурмилла', 'Девон-рекс', 'Европейская короткошерстная',
|
4 |
+
'Экзотическая короткошерстная', 'Мейн-кун', 'Нибелунг',
|
5 |
+
'Персидская', 'Питерболд', 'Рэгдолл', 'Русская голубая',
|
6 |
+
'Саванна', 'Шотландская вислоухая', 'Сибирская', 'Сингапурская',
|
7 |
+
'Сомалийская', 'Сфинкс', 'Черепаховая'
|
8 |
+
]
|
9 |
+
|
10 |
+
CAT_DESCRIPTIONS = {
|
11 |
+
'Бенгальская':
|
12 |
+
'''# Бенгальская кошка
|
13 |
+
### Бенгальская кошка — крупная кошка с блестящей \
|
14 |
+
шерстью, очень развитой мускулатурой и толстым \
|
15 |
+
хвостом, который часто свисает до земли. Бенгальская \
|
16 |
+
кошка внешне напоминает дикую кошку: характерная \
|
17 |
+
густая роскошная шерсть имеет пятнистый окрас и \
|
18 |
+
мраморные разводы. На широкой голове располагаются \
|
19 |
+
небольшие уши и бакенбарды, глаза имеют \
|
20 |
+
миндалевидную форму и чёрную окантовку. Хвост \
|
21 |
+
толстый, сужается к чёрному кончику. Ни одна другая \
|
22 |
+
порода кошек не имеет такого эффекта золотистого \
|
23 |
+
или жемчужного мерцания шерсти, который можно \
|
24 |
+
обнаружить у бенгальских кошек.
|
25 |
+
''',
|
26 |
+
|
27 |
+
'Бомбейская':
|
28 |
+
'''# Бомбейская кошка
|
29 |
+
### Бомбейская кошка (бомбей) – питомец-компаньон \
|
30 |
+
для активных и жизнерадостных людей, проводящих \
|
31 |
+
большую часть свободного времени дома. Порода \
|
32 |
+
задумана как мини-версия черной пантеры. \
|
33 |
+
Однако от хищной кошки унаследовала только экстерьер. \
|
34 |
+
В то время как характер больше напоминает собаку. \
|
35 |
+
Породистый бомбей демонстрирует безусловную преданность \
|
36 |
+
и любовь к человеку, не отказывается от игры в апорт и \
|
37 |
+
легко запоминает базовые команды. К тому же \
|
38 |
+
представители породы умны и эмпатичны: животное тонко \
|
39 |
+
чувствует настроение хозяина и подстраивается даже под \
|
40 |
+
незначительные его изменения.
|
41 |
+
''',
|
42 |
+
|
43 |
+
'Британская короткошерстная':
|
44 |
+
'''# Британская короткошерстная
|
45 |
+
### Британская короткошёрстная — крупная порода, \
|
46 |
+
имеющая массивное, коренастое телосложение. У \
|
47 |
+
представителей этой породы широкая грудная клетка \
|
48 |
+
и толстый хвост. Морда, уши, глаза и голова — имеют \
|
49 |
+
округлую форму. У кошки короткий широкий нос и \
|
50 |
+
массивная нижняя челюсть. Шерсть у британских \
|
51 |
+
короткошёрстных — короткая и густая, без пушистости, \
|
52 |
+
бывает более ста окрасов и имеет множество \
|
53 |
+
комбинаций пятен.
|
54 |
+
''',
|
55 |
+
|
56 |
+
'Бурмилла':
|
57 |
+
'''# Бурмилла
|
58 |
+
### Бурмилла — кошка среднего телосложения, похожая \
|
59 |
+
на бурмезскую. Порода отличается коренастым и \
|
60 |
+
довольно мускулистым телом и широкой прямой спиной. \
|
61 |
+
Самки намного мельче и изящнее самцов. Форма головы — \
|
62 |
+
слегка скругленный конус. Представители этой породы \
|
63 |
+
имеют клиновидный нос и широко посаженные уши. Цвет глаз \
|
64 |
+
варьируется от золотисто-желтого до зеленого. \
|
65 |
+
Отличительной особенностью этой породы является \
|
66 |
+
характерная отметина на лбу в виде буквы M. Короткая \
|
67 |
+
шерсть со светлым подшерстком близко прилегает к телу \
|
68 |
+
и отличается равномерным типпингом(кончики волосков более \
|
69 |
+
темного цвета) по всему телу.
|
70 |
+
''',
|
71 |
+
|
72 |
+
'Девон-рекс':
|
73 |
+
'''# Девон-рекс
|
74 |
+
### Запоминающийся образ кошки породы девон-рекс — это \
|
75 |
+
ехидная мордочка с волнистой «мятой» шерстью. Широкие \
|
76 |
+
скулы, большие глаза, низко посаженные уши. Короткая \
|
77 |
+
шерсть имеет волнистые остевые волосы, но в основном \
|
78 |
+
состоит из пухового подшерстка. Шерсть девон-рекса \
|
79 |
+
имеет характерную «марсельскую волну», особенно на спине, \
|
80 |
+
где образуются мелкие барашки. Бакенбарды и надбровные \
|
81 |
+
дуги у этой породы также волнистые. Девон-рекс может \
|
82 |
+
быть любых окрасов и вариантов пятнистости.
|
83 |
+
''',
|
84 |
+
|
85 |
+
'Европейская короткошерстная':
|
86 |
+
'''# Европейская короткошерстная
|
87 |
+
### Европейская короткошерстная кошка, она же кельтская, \
|
88 |
+
на первый взгляд выглядит как обычная довольно крупная \
|
89 |
+
домашняя кошка. Тем не менее генетики выяснили, что она \
|
90 |
+
наследует кошкам, охранявшим амбары европейских крестьян \
|
91 |
+
2000 лет назад! И это как минимум. Каждая кельтская кошка \
|
92 |
+
отличаются своим характером: одни любят прогулки, другие \
|
93 |
+
предпочитают наблюдать жизнь из окна; одни более \
|
94 |
+
общительны, другие менее. Но все они хорошие компаньоны: \
|
95 |
+
уравновешенные, жизнерадостные, ласковые без навязчивости, \
|
96 |
+
готовые и поиграть, и просто составить компанию на диване \
|
97 |
+
перед телевизором.
|
98 |
+
''',
|
99 |
+
|
100 |
+
'Экзотическая короткошерстная':
|
101 |
+
'''# Экзотическая короткошерстная
|
102 |
+
### Экзотическая короткошерстная кошка среднего размера. \
|
103 |
+
Телосложение — коренастое и в то же время подтянутое. \
|
104 |
+
Имеет широкую, круглую голову с небольшими, широко \
|
105 |
+
посаженными ушами. У этой кошки маленькая открытая \
|
106 |
+
мордочка с большими яркими глазами. Имеет короткие и \
|
107 |
+
сильные лапы, с крупными округлыми стопами. Хвост \
|
108 |
+
небольшой длины, при этом достаточно пушистый. Шерсть \
|
109 |
+
— короткая, густая, с плотным мягким подшерстком. \
|
110 |
+
Экзотическая короткошерстная кошка бывает многих \
|
111 |
+
окрасов и вариантов пятнистости, типичных для персидской \
|
112 |
+
породы, а также может иметь окрас «пятнистый табби».
|
113 |
+
''',
|
114 |
+
|
115 |
+
'Мейн-кун':
|
116 |
+
'''# Мейн-кун
|
117 |
+
### Мейн-кун – это популярная порода кошек, завоевавшая \
|
118 |
+
поклонников по всему миру своим харизматичным внешним \
|
119 |
+
видом и внушительными размерами. Даже обычным людям, \
|
120 |
+
совсем не разбирающимся в породах, легко отличить \
|
121 |
+
мейн-куна от других кошек благодаря их выдающимся \
|
122 |
+
внешним данным. Мейн-кун гордо носит звание пушистого \
|
123 |
+
великана.
|
124 |
+
''',
|
125 |
+
|
126 |
+
'Нибелунг':
|
127 |
+
'''# Нибелунг
|
128 |
+
### Нибелунг — редкая порода домашних кошек. Она \
|
129 |
+
считается длинношёрстным вариантом русской голубой \
|
130 |
+
кошки. Представители этой породы замкнуты и не \
|
131 |
+
«горласты», но иногда могут громко мяукать дожидаясь \
|
132 |
+
хозяина. Обладают не требующей ежедневного ухода \
|
133 |
+
голубой шерстью с серебристым типпингом. Остевые \
|
134 |
+
волосы шерсти нибелунга отражают свет, тем самым \
|
135 |
+
создавая серебристый отлив. Окрас шерсти и подшёрстка \
|
136 |
+
— сплошной голубой. Порода названа в честь «порождения \
|
137 |
+
тумана» (от нем. Nebel — туман) из средневекового \
|
138 |
+
немецкого эпоса «Песнь о Нибелунгах».
|
139 |
+
''',
|
140 |
+
|
141 |
+
'Персидская':
|
142 |
+
'''# Персидская
|
143 |
+
### Персидская длинношерстная кошка — порода среднего \
|
144 |
+
размера, с коренастым телом и с массивной головой. \
|
145 |
+
Уши — небольшие, широко посаженные. У этой кошки \
|
146 |
+
маленькая открытая мордочка с большими яркими глазами. \
|
147 |
+
Имеет короткие и сильные лапы с крупными округлыми \
|
148 |
+
основаниями и кисточками между подушечек. Шерсть — \
|
149 |
+
длинная, густая и блестящая, с плотным мягким \
|
150 |
+
подшерстком. Имеет короткий, но необыкновенно пушистый \
|
151 |
+
хвост. Допускаются разные варианты окрасов и \
|
152 |
+
пятнистости.
|
153 |
+
''',
|
154 |
+
|
155 |
+
'Питерболд':
|
156 |
+
'''# Питерболд
|
157 |
+
### Питерболд — выведенная в России порода бесшёрстных \
|
158 |
+
домашних кошек. Питерболды имеют характерную форму \
|
159 |
+
головы: длинную и ��зкую, с прямым профилем, \
|
160 |
+
миндалевидными глазами и большими, разведёнными в \
|
161 |
+
стороны ушами. Идеальные представители породы выглядят, \
|
162 |
+
как бесшёрстные ориентальные кошки. Питерболды обычно \
|
163 |
+
дружелюбны, любопытны, активны, любят всех членов своей \
|
164 |
+
семьи и нуждаются в общении с людьми; они не принадлежат \
|
165 |
+
к независимым кошкам. Питерболды хорошо уживаются с \
|
166 |
+
другими кошками и прочими домашними животными.
|
167 |
+
''',
|
168 |
+
|
169 |
+
'Рэгдолл':
|
170 |
+
'''# Рэгдолл
|
171 |
+
### Рэгдолл — крупная и величавая кошка с характерным \
|
172 |
+
окрасом. Представители этой породы обычно расслаблены, \
|
173 |
+
невозмутимы и кажутся ленивыми. Голова широкая, имеет \
|
174 |
+
плоскую верхнюю часть, с широко расположенными ушами. \
|
175 |
+
У рэгдолла большие глаза насыщенного голубого цвета. \
|
176 |
+
Тело — длинное и мускулистое, с широкой грудной клеткой, \
|
177 |
+
короткой шеей и крупными круглыми лапами с кисточками \
|
178 |
+
между подушечек. Хвост — длинный и пушистый. Шерсть \
|
179 |
+
рэгдоллов шелковистая и густая, средней длины или \
|
180 |
+
длинная. Для взрослых кошек типично «кольцо» шерсти \
|
181 |
+
вокруг шеи и пушистые «штанишки». Встречаются три вида \
|
182 |
+
окраса: колор-пойнт (у таких кошек светлое тело, но \
|
183 |
+
тёмные отдельные его части: лапы, хвост, уши и морда), \
|
184 |
+
би-колор (один из типов окрасов кошек с белыми пятнами, \
|
185 |
+
которые занимают от 1/3 до 1/2 всего тела) и миттед \
|
186 |
+
(один из типов окрасов кошек с белыми пятнами, \
|
187 |
+
которые занимают от 1/4 до 1/3 всего тела).
|
188 |
+
''',
|
189 |
+
|
190 |
+
'Русская голубая':
|
191 |
+
'''# Русская голубая
|
192 |
+
### Кошки породы русская голубая бывают среднего или \
|
193 |
+
крупного размера. Имеют изящное тело и длинные, \
|
194 |
+
стройные ноги. Эта кошка настолько грациозна, что \
|
195 |
+
кажется, будто она передвигается на цыпочках. Голова \
|
196 |
+
имеет клинообразную форму с выдающейся зоной бакенбард \
|
197 |
+
и крупные уши. Яркие зеленые глаза широко посажены и \
|
198 |
+
имеют миндалевидную форму. По текстуре шерсть русской \
|
199 |
+
голубой непохожа на шерсть кошек других пород, и это \
|
200 |
+
основная отличительная особенность породы. Шерстка у этих \
|
201 |
+
животных — двойная, с густым подшерстком, короткая и \
|
202 |
+
приятная на ощупь. Несмотря на название породы, иногда \
|
203 |
+
среди русских голубых кошек встречаются особи с черным \
|
204 |
+
и белым окрасом. У типичных же представителей окрас \
|
205 |
+
равномерно голубой, с серебристым отливом.
|
206 |
+
''',
|
207 |
+
|
208 |
+
'Саванна':
|
209 |
+
'''# Саванна
|
210 |
+
### Саванна — высокая худая кошка с длинными лапами. \
|
211 |
+
Взрослая особь может достигать веса в 15 кг при росте \
|
212 |
+
60 см в холке. Размеры ее тела зависят от близости \
|
213 |
+
конкретного представителя к первому скрещенному \
|
214 |
+
поколению: чем ближе, тем крупнее животное. Кроме \
|
215 |
+
того, под влиянием случайных факторов в генетике породы \
|
216 |
+
саванна могут встречаться кошки, значительно \
|
217 |
+
отличающиеся по размерам, даже в одном помете. Как \
|
218 |
+
правило, саванна имеет пятнистый окрас, но иногда \
|
219 |
+
встречаются кошки с мраморным рисунком, снежным окрасом, \
|
220 |
+
а также с шерстью голубого или других ненасыщенных цветов. \
|
221 |
+
Уши — высокие, с глубокой «чашкой», округленные, прямые \
|
222 |
+
(задняя часть уха имеет светлую полосу по центру, с \
|
223 |
+
черным, темно-серым или коричневым окаймлением, что \
|
224 |
+
создает «эффект глаза»). У этих кошек широкий нос и глаза \
|
225 |
+
с нависающими веками. На коротких хвостах — черные кольца \
|
226 |
+
и однотонный черный кончик. Глаза могут быть зелеными, \
|
227 |
+
коричневыми или золотистыми.
|
228 |
+
''',
|
229 |
+
|
230 |
+
'Шотландская вислоухая':
|
231 |
+
'''# Шотландская вислоухая
|
232 |
+
### Шотландская вислоухая кошка или скоттиш-фолд \
|
233 |
+
(англ. Scottish Fold) — порода домашних кошек с \
|
234 |
+
характерными загнутыми вперёд и вниз ушами, что \
|
235 |
+
вызвано действием доминантного гена, который возник \
|
236 |
+
в результате естественной мутации и оказывающий \
|
237 |
+
влияние на хрящи во всём организме. Это придаёт \
|
238 |
+
голове кошки характерный вид, которую часто описывают \
|
239 |
+
как «похожую на совиную».
|
240 |
+
''',
|
241 |
+
|
242 |
+
'Сибирская':
|
243 |
+
'''# Сибирская
|
244 |
+
### Сибирская кошка – прекрасный выбор для семей с детьми \
|
245 |
+
и владельцев, которые интересуются разведением и \
|
246 |
+
выставочной деятельностью. Ведь порода обладает устойчивой \
|
247 |
+
психикой и крепким здоровьем, при этом все еще открыта \
|
248 |
+
для сертификации. Разводить «сибиряков» можно даже с \
|
249 |
+
беспородной кошкой, обладающей характерными признаками \
|
250 |
+
представителей породы. Поэтому привнесение новой крови \
|
251 |
+
в племенную линию осуществляется регулярно, и порода \
|
252 |
+
не страдает от инбридинга (скрещивания особей, состоящих \
|
253 |
+
в тесном родстве).
|
254 |
+
''',
|
255 |
+
|
256 |
+
'Сингапурская':
|
257 |
+
'''# Сингапурская
|
258 |
+
### Сингапурская кошка похожа на маленького представителя \
|
259 |
+
абиссинской породы. Ее голова имеет округлую форму с \
|
260 |
+
выраженными бакенбардами и короткой широкой мордой. \
|
261 |
+
Уши — крупные и глубокие. Глаза — большие, мин��алевидной \
|
262 |
+
формы, слегка скошенные, широко раскрытые. Несмотря на \
|
263 |
+
небольшой размер, тело кошки — коренастое и мускулистое. \
|
264 |
+
Хвост — тонкий с тупым кончиком. Шерсть — короткая и \
|
265 |
+
гладкая, с черными, коричневыми или кремовыми полосками \
|
266 |
+
на каждом из волосков. При этом шерсть имеет более \
|
267 |
+
темный окрас на спине, хвосте и ногах, а на груди и \
|
268 |
+
животе остается светлой. Глаза у кошек этой породы \
|
269 |
+
обычно карие, зеленые или желтые с черным окаймлением.
|
270 |
+
''',
|
271 |
+
|
272 |
+
'Сомалийская':
|
273 |
+
'''# Сомалийская
|
274 |
+
### Сомалийская кошка имеет среднее телосложение. Это \
|
275 |
+
крепкое, стройное животное с хорошо развитой \
|
276 |
+
мускулатурой и шерстью средней длины. Голова кошки \
|
277 |
+
имеет форму клина с мягкими очертаниями. Уши широко \
|
278 |
+
посаженные, с кисточками. Имеет глаза миндалевидной \
|
279 |
+
формы, часто — янтарного цвета, оттенка грецкого ореха \
|
280 |
+
или зеленые. . Взрослая сомалийская кошка имеет \
|
281 |
+
«кольцо» шерсти вокруг шеи и пушистые «штанишки», \
|
282 |
+
что не выражено у котят. Мягкие, красивые, густые \
|
283 |
+
волоски плотно прилегают к телу, при этом каждый из \
|
284 |
+
них окрашен как минимум в три цвета. Сомалийская кошка \
|
285 |
+
имеет 28 окрасов. Наиболее популярный — насыщенный \
|
286 |
+
золотисто-коричневый с кремовой основой и черным \
|
287 |
+
тикингом.
|
288 |
+
''',
|
289 |
+
|
290 |
+
'Сфинкс':
|
291 |
+
'''# Сфинкс
|
292 |
+
### Сфинкс – бесшерстная кошка с поведением и характером \
|
293 |
+
собаки. Представители породы относятся к классу \
|
294 |
+
компаньонов. Поэтому нуждаются во внимании и совсем \
|
295 |
+
непохожи на независимых кошек, стремящихся держаться \
|
296 |
+
от человека на расстоянии. Животное ориентировано на \
|
297 |
+
взаимодействие, любит сидеть на коленях, легко уживается \
|
298 |
+
с другими животными в доме и хорошо ладит с детьми, даже \
|
299 |
+
самыми маленькими. Порода сфинкс имеет несколько \
|
300 |
+
ответвлений. Например, канадский сфинкс, донской сфинкс \
|
301 |
+
и петерболд. Однако признание от международных \
|
302 |
+
фелинологических организаций заслужил пока лишь канадский.
|
303 |
+
''',
|
304 |
+
|
305 |
+
'Черепаховая':
|
306 |
+
'''# Черепаховая
|
307 |
+
### Черепаховый окрас – один из самых удивительных и \
|
308 |
+
необычных окрасов кошек. Черепаховый окрас является \
|
309 |
+
очень распространённым среди метисов, а также среди \
|
310 |
+
многих пород кошек он закреплён в стандарты. Вариации \
|
311 |
+
цветов полосок и пятен подобного окраса, а также их \
|
312 |
+
узоры на шерсти уникальны и неповторимы, так же, как и \
|
313 |
+
отпечатки пальцев у людей. Это генетическая особенность, \
|
314 |
+
которая передается по наследству и обуславливается \
|
315 |
+
определенной комбинацией хромосом.
|
316 |
+
'''
|
317 |
+
}
|
main_app.py
CHANGED
@@ -28,18 +28,27 @@ def predict(image, api_mode=False):
|
|
28 |
predicted_breed = CAT_BREEDS[np.argmax(prediction)]
|
29 |
breed_description = CAT_DESCRIPTIONS[predicted_breed]
|
30 |
|
|
|
|
|
31 |
if api_mode:
|
32 |
-
|
33 |
-
|
34 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
35 |
|
36 |
|
37 |
@app.post('/predict_breed/')
|
38 |
def predict_api(url: Url):
|
39 |
try:
|
40 |
image = requests.get(url.link).content
|
41 |
-
except:
|
42 |
-
return {'error': 'Invalid link'}
|
43 |
image = Image.open(BytesIO(image))
|
44 |
return predict(image, api_mode=True)
|
45 |
|
|
|
28 |
predicted_breed = CAT_BREEDS[np.argmax(prediction)]
|
29 |
breed_description = CAT_DESCRIPTIONS[predicted_breed]
|
30 |
|
31 |
+
all_predictions = {CAT_BREEDS[i]: float(prediction[i]) for i in range(20)}
|
32 |
+
|
33 |
if api_mode:
|
34 |
+
breed_description = ' '.join(breed_description.replace('\n', '.')
|
35 |
+
.replace('#', '')
|
36 |
+
.split())
|
37 |
+
|
38 |
+
return {
|
39 |
+
'breed': predicted_breed,
|
40 |
+
'description': breed_description,
|
41 |
+
'predictions': all_predictions
|
42 |
+
}
|
43 |
+
return all_predictions, breed_description
|
44 |
|
45 |
|
46 |
@app.post('/predict_breed/')
|
47 |
def predict_api(url: Url):
|
48 |
try:
|
49 |
image = requests.get(url.link).content
|
50 |
+
except Exception as e:
|
51 |
+
return {'error': 'Invalid link', 'exception': str(e)}
|
52 |
image = Image.open(BytesIO(image))
|
53 |
return predict(image, api_mode=True)
|
54 |
|