Update app.py

app.py

@@ -7,10 +7,10 @@ model_name = "allenai/scibert_scivocab_uncased"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSequenceClassification.from_pretrained(model_name, num_labels=3)
 
-# 2️⃣ Dataset laden (armanc/scientific_papers) mit trust_remote_code=True
-dataset = load_dataset("armanc/scientific_papers", trust_remote_code=True)
+# 2️⃣ Dataset laden (mit spezifischer Konfiguration: "arxiv" oder "pubmed")
+dataset = load_dataset("armanc/scientific_papers", "arxiv", trust_remote_code=True)  # Oder "pubmed"
 
-# 3️⃣ Tokenisierung der Texte (hier wird die Spalte "text" genutzt; ggf. anpassen, falls andere Spalten vorhanden sind)
+# 3️⃣ Tokenisierung der Texte
 def tokenize_function(examples):
     return tokenizer(examples["text"], padding="max_length", truncation=True)