Update app.py

app.py

@@ -1,12 +1,61 @@
 import gradio as gr
+from transformers import AutoModel
 
-def greet(name, intensity):
-    return "Hello, " + name + "!" * int(intensity)
+model = AutoModel.from_pretrained("SnowFlash383935/DigitalEduTransformers", trust_remote_code=True)
+
+def greet(sex: int, has_photo: bool, has_mobile: bool, followers_count: int,
+          relation: int, life_main: int, people_main: int):
+    return model([[sex, has_photo, has_mobile, followers_count,
+                 relation, life_main, people_main]])[0]
+
+# Создаем выпадающие списки с вариантами ответов
+RELATION_OPTIONS = [
+    "Не указано",
+    "Не женат/не замужем",
+    "Есть друг/подруга",
+    "Помолвлен(а)",
+    "Женат/замужем",
+    "Всё сложно",
+    "В активном поиске",
+    "Влюблен(а)",
+    "В гражданском браке"
+]
+
+LIFE_MAIN_OPTIONS = [
+    "Не указано",
+    "Семья и дети",
+    "Карьера и деньги",
+    "Развлечения и отдых",
+    "Наука и исследования",
+    "Совершенствование мира",
+    "Саморазвитие",
+    "Красота и искусство",
+    "Слава и влияние"
+]
+
+PEOPLE_MAIN_OPTIONS = [
+    "Не указано",
+    "Ум и креативность",
+    "Доброта и честность",
+    "Красота и здоровье",
+    "Власть и богатство",
+    "Смелость и упорство",
+    "Юмор и жизнелюбие"
+]
 
 demo = gr.Interface(
     fn=greet,
-    inputs=["text", "slider"],
-    outputs=["text"],
+    inputs=[
+        gr.Radio(["Женщина", "Мужчина"], type="index", label="Пол", value="Женщина", ),
+        gr.Checkbox(label="Есть фото профиля"),
+        gr.Checkbox(label="Указан мобильный телефон"),
+        gr.Number(label="Количество подписчиков", precision=0),
+        gr.Dropdown(RELATION_OPTIONS, type="index", label="Семейное положение"),
+        gr.Dropdown(LIFE_MAIN_OPTIONS, type="index", label="Главное в жизни"),
+        gr.Dropdown(PEOPLE_MAIN_OPTIONS, type="index", label="Главное в людях"),
+    ],
+    outputs=[gr.Checkbox(label="Купит ли курс")],
+    title="Прогнозирование покупки курса",
 )
 
 demo.launch()