orzhan commited on
Commit
053a005
1 Parent(s): aa1149a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -11,10 +11,8 @@ model_name = "orzhan/ruroberta-ruatd-binary"
11
  @st.cache
12
  def load_model(model_name):
13
  tokenizer=AutoTokenizer.from_pretrained(tokenizer_name)
14
- model = AutoModel.from_pretrained(model_name)
15
-
16
- pipeline(task="text-classification",model=model_name)
17
- odel
18
 
19
  model = load_model(model_name)
20
 
 
11
  @st.cache
12
  def load_model(model_name):
13
  tokenizer=AutoTokenizer.from_pretrained(tokenizer_name)
14
+ model = AutoModel.from_pretrained(model_name)
15
+ return pipeline(task="text-classification",model=model,tokenizer=tokenizer)
 
 
16
 
17
  model = load_model(model_name)
18