Update app.py

app.py

@@ -16,13 +16,14 @@ def classify(item: str, categories: list) -> str:
     result = classifier(prompt, truncation=True)
     print(result) 
     label_index = int(result[0]['label'].split("_")[1])
-    return CATEGORIES[label_index]
+    #return CATEGORIES[label_index]
+    return result
 
 # Интерфейс Gradio
 iface = gr.Interface(
     fn=classify,
     inputs=[
-        gr.Textbox(label="Название товара cointegrated/rubert-tiny2 +"), 
+        gr.Textbox(label="Название товара cointegrated/rubert-tiny2 ++"), 
         gr.Textbox(label="Категории (через запятую)", value="Техника, Овощи, Инструменты, Коробки, Прочее")
     ],
     outputs=gr.Textbox(label="Категория")