Update app.py

app.py

@@ -18,13 +18,13 @@ load_face2paint_model(512)  # Загрузка модели с размером
 def inference(img, ver, size):
     if size != 512:
         load_face2paint_model(size)  # Загрузка модели с выбранным размером
-    if ver == 'Осветлённый':
+    if ver == 'Стиль - 1':
         out = face2paint(model1, img)
-    elif ver == 'Мужчина':
+    elif ver == 'Стиль - 2':
         out = face2paint(model2, img)
-    elif ver == 'Женщина':
+    elif ver == 'Стиль - 3':
         out = face2paint(model3, img)
-    elif ver == 'Паприка':
+    elif ver == 'Стиль - 4':
         out = face2paint(model4, img)
     return out
 title = "Фото в Рисунок"
@@ -33,10 +33,10 @@ article = "<br><br><br><br><br>"
 # Создание интерфейса Gradio с вариантами моделей, стилей и размеров и черным фоном
 interface = gr.Interface(inference,
                          [gr.inputs.Image(type="pil"),
-                          gr.inputs.Radio(['Осветлённый', 'Мужчина', 'Женщина', 'Паприка'],
+                          gr.inputs.Radio(['Стиль - 1', 'Стиль - 2', 'Стиль - 3', 'Стиль - 4'],
                                          type="value",
-                                         default='Осветлённый',
-                                         label='Имя модели'),
+                                         default='Стиль - 1',
+                                         label='Выбор стиля'),
                           gr.inputs.Radio([512, 1024],
                                          type="value",
                                          default=512,