SocialScrape commited on
Commit
3376064
·
verified ·
1 Parent(s): dce6892

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -1,11 +1,15 @@
 
1
  import torch
2
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
3
  import gradio as gr
4
 
5
- MODEL_NAME = "SocialScrape/longformer-my-classifier" # твоя модель на Hugging Face
6
 
7
- tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
8
- model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME)
 
 
 
9
 
10
  def classify_text(text):
11
  inputs = tokenizer(text, return_tensors="pt", truncation=True)
 
1
+ import os
2
  import torch
3
  from transformers import AutoTokenizer, AutoModelForSequenceClassification
4
  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
+ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, token=hf_token)
12
+ model = AutoModelForSequenceClassification.from_pretrained(MODEL_NAME, token=hf_token)
13
 
14
  def classify_text(text):
15
  inputs = tokenizer(text, return_tensors="pt", truncation=True)