Yehor commited on
Commit
5e2b13a
·
verified ·
1 Parent(s): 75ed04e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,9 +2,9 @@ import gradio as gr
2
 
3
  from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
4
 
5
- tokenizer = AutoTokenizer.from_pretrained("ua-l/topics-classifier")
6
  model = AutoModelForSequenceClassification.from_pretrained(
7
- "ua-l/topics-classifier"
8
  )
9
 
10
  topic_classifier = pipeline(
 
2
 
3
  from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
4
 
5
+ tokenizer = AutoTokenizer.from_pretrained("ukr-models/xlm-roberta-base-uk")
6
  model = AutoModelForSequenceClassification.from_pretrained(
7
+ "ua-l/topics-classifier-xlm-roberta-base-uk-v2"
8
  )
9
 
10
  topic_classifier = pipeline(