SocialScrape commited on
Commit
ca4ac56
·
verified ·
1 Parent(s): cf1ab11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -5,14 +5,14 @@ import gradio as gr
5
 
6
  MODEL_NAME = "SocialScrape/longformer-my-classifier"
7
 
8
- # Бере токен із секретів Hugging Face Space
9
  hf_token = os.getenv("HF_TOKEN")
10
 
11
  # Завантаження токенізатора та моделі
12
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, token=hf_token)
13
  model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME, token=hf_token)
14
 
15
- # Функція класифікації тексту
16
  def classify_text(text):
17
  inputs = tokenizer(text, return_tensors="pt", truncation=True)
18
  outputs = model(**inputs)
@@ -24,10 +24,13 @@ def classify_text(text):
24
  "Ймовірності класів": probabilities.tolist()
25
  }
26
 
27
- # Інтерфейс Gradio
28
  iface = gr.Interface(
29
  fn=classify_text,
30
  inputs=gr.Textbox(label="Введіть текст для класифікації"),
31
  outputs=gr.JSON(label="Результати класифікації"),
32
  title="Longformer UK Text Classifier"
33
- )
 
 
 
 
5
 
6
  MODEL_NAME = "SocialScrape/longformer-my-classifier"
7
 
8
+ # Бере токен із секретів Hugging Face Space (у Settings -> Secrets)
9
  hf_token = os.getenv("HF_TOKEN")
10
 
11
  # Завантаження токенізатора та моделі
12
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, token=hf_token)
13
  model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME, token=hf_token)
14
 
15
+ # Основна функція класифікації
16
  def classify_text(text):
17
  inputs = tokenizer(text, return_tensors="pt", truncation=True)
18
  outputs = model(**inputs)
 
24
  "Ймовірності класів": probabilities.tolist()
25
  }
26
 
27
+ # Градіо інтерфейс
28
  iface = gr.Interface(
29
  fn=classify_text,
30
  inputs=gr.Textbox(label="Введіть текст для класифікації"),
31
  outputs=gr.JSON(label="Результати класифікації"),
32
  title="Longformer UK Text Classifier"
33
+ )
34
+
35
+ # Обов'язково запускати
36
+ iface.launch()