hh1199 commited on
Commit
7ab3fbc
·
verified ·
1 Parent(s): 9a6d831

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +45 -0
app.py CHANGED
@@ -0,0 +1,45 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ from transformers import pipeline
3
+ import time
4
+
5
+ # Загружаем модель ПЕРВОЙ
6
+ try:
7
+ classifier = pipeline(
8
+ "text-classification",
9
+ model="cointegrated/rubert-tiny2",
10
+ truncation=True
11
+ )
12
+ except Exception as e:
13
+ print(f"Ошибка загрузки модели: {e}")
14
+ raise
15
+
16
+ # Функция классификации
17
+ def classify(item: str, categories: str) -> str:
18
+ try:
19
+ prompt = f"Отнеси '{item}' к категории: {categories}. Ответь только категорией."
20
+ result = classifier(prompt)
21
+ return result[0]['label']
22
+ except Exception as e:
23
+ return f"Ошибка: {str(e)}"
24
+
25
+ # Создаем интерфейс ПОСЛЕ загрузки модели
26
+ iface = gr.Interface(
27
+ fn=classify,
28
+ inputs=[
29
+ gr.Textbox(label="Товар"),
30
+ gr.Textbox(label="Категории (через запятую)", value="Овощи, Инструменты")
31
+ ],
32
+ outputs="text",
33
+ title="Классификатор"
34
+ )
35
+
36
+ # Явно запускаем сервер
37
+ iface.launch(
38
+ server_name="0.0.0.0",
39
+ server_port=7860,
40
+ show_error=True
41
+ )
42
+
43
+ # Бесконечный цикл, чтобы сервер не завершался
44
+ while True:
45
+ time.sleep(60)