Spaces:
Runtime error
Runtime error
import torch | |
from transformers import AutoTokenizer, AutoModel | |
# Modell & Tokenizer laden | |
model_name = "allenai/scibert_scivocab_uncased" | |
tokenizer = AutoTokenizer.from_pretrained(model_name) | |
model = AutoModel.from_pretrained(model_name) | |
# Explizit auf CPU setzen | |
device = torch.device("cpu") | |
model.to(device) | |
# Beispiel-Text für den Test | |
text = "This paper introduces a novel deep learning approach for cancer diagnosis." | |
# Tokenisierung & Modell-Durchlauf | |
inputs = tokenizer(text, return_tensors="pt").to(device) | |
outputs = model(**inputs) | |
print("Modell erfolgreich geladen!") | |
print("Shape der Ausgabe:", outputs.last_hidden_state.shape) |