app.py

import os
import torch
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import gradio as gr

MODEL_NAME = "SocialScrape/longformer-my-classifier"

# Бере токен із секретів Hugging Face Space
hf_token = os.getenv("HF_TOKEN")

tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, token=hf_token)
model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME, token=hf_token)

def classify_text(text):
    inputs = tokenizer(text, return_tensors="pt", truncation=True)
    outputs = model(**inputs)
    probabilities = torch.softmax(outputs.logits, dim=-1)[0].detach().numpy()
    predicted_class = probabilities.argmax()

    return {
        "Клас": int(predicted_class),
        "Ймовірності класів": probabilities.tolist()
    }

iface = gr.Interface(
    fn=classify_text,
    inputs=gr.Textbox(label="Введіть текст для класифікації"),
    outputs=gr.JSON(label="Результати класифікації"),
    title="Longformer UK Text Classifier"
)

if __name__ == "__main__":
    iface.launch()