flytoe commited on
Commit
3e7c541
·
verified ·
1 Parent(s): 1f3546c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -2
app.py CHANGED
@@ -1,3 +1,21 @@
1
  import torch
2
- print(torch.__version__)
3
- print(torch.cuda.is_available()) # Zeigt, ob eine GPU erkannt wird
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import torch
2
+ from transformers import AutoTokenizer, AutoModel
3
+
4
+ # Modell & Tokenizer laden
5
+ model_name = "allenai/scibert_scivocab_uncased"
6
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
7
+ model = AutoModel.from_pretrained(model_name)
8
+
9
+ # Explizit auf CPU setzen
10
+ device = torch.device("cpu")
11
+ model.to(device)
12
+
13
+ # Beispiel-Text für den Test
14
+ text = "This paper introduces a novel deep learning approach for cancer diagnosis."
15
+
16
+ # Tokenisierung & Modell-Durchlauf
17
+ inputs = tokenizer(text, return_tensors="pt").to(device)
18
+ outputs = model(**inputs)
19
+
20
+ print("Modell erfolgreich geladen!")
21
+ print("Shape der Ausgabe:", outputs.last_hidden_state.shape)