Hugging Face's logo

Spaces:
philipp-zettl
/
multi-head-classification
Runtime error

App Files Community
philipp-zettl commited on
Commit
fd2392a
·
verified ·
1 Parent(s): 1005357

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -3,7 +3,10 @@ from transformers import AutoTokenizer
3
  from model import MultiHeadClassification
4
 
5
 
6
- model = MultiHeadClassification.from_pretrained('philipp-zettl/multi-head-sequence-classification-model')
 
 
 
7
  tokenizer = AutoTokenizer.from_pretrained('BAAI/bge-m3')
8
 
9
  def generate(prompt):
 
3
  from model import MultiHeadClassification
4
 
5
 
6
+ model = MultiHeadClassification.from_pretrained(
7
+ 'philipp-zettl/multi-head-sequence-classification-model',
8
+ {"GGU": 3, "sentiment": 3}
9
+ )
10
  tokenizer = AutoTokenizer.from_pretrained('BAAI/bge-m3')
11
 
12
  def generate(prompt):