flytoe commited on
Commit
def161a
·
verified ·
1 Parent(s): a800181

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -13
app.py CHANGED
@@ -1,15 +1,6 @@
1
  from transformers import AutoTokenizer, AutoModel
2
 
3
- # Laden des SciBERT-Modells
4
- tokenizer = AutoTokenizer.from_pretrained('allenai/scibert_scivocab_uncased')
5
- model = AutoModel.from_pretrained('allenai/scibert_scivocab_uncased')
6
-
7
- # Beispiel-Text
8
- text = "This paper introduces a novel deep learning approach for cancer diagnosis."
9
-
10
- # Tokenisierung & Modell-Durchlauf
11
- inputs = tokenizer(text, return_tensors="pt") # Falls du TensorFlow nutzt: return_tensors="tf"
12
- outputs = model(**inputs)
13
-
14
- # Anzeigen der Ausgabe
15
- print(outputs.last_hidden_state.shape)
 
1
  from transformers import AutoTokenizer, AutoModel
2
 
3
+ # Explizite Angabe von torch_dtype kann helfen
4
+ model_name = "allenai/scibert_scivocab_uncased"
5
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
6
+ model = AutoModel.from_pretrained(model_name, torch_dtype=torch.float32)