Update app.py

app.py

@@ -52,11 +52,8 @@ def compute_metrics(eval_pred):
 #Access-Token (in Secrets)
 
 #aus den Secrets importieren (siehe Setting zu diesem Space)
-xxx = os.environ["HF_ACCESS_TOKEN"]
-
-#dotenv.config({ path: ".env.local" });
-#login(token=process.env.HF_ACCESS_TOKEN)
-login(token=xxx)
+login(token=os.environ["HF_ACCESS_TOKEN"])
+     
 #Modelle und Tokenizer
 
 #Alternativ mit beliebigen Modellen:
@@ -138,6 +135,20 @@ print("trained!!!!!")
 #trainer.push_to_hub("test-tis", use_auth_token=True)
 #trainer.push_to_hub()
 
+############################
+#Test
+############################
+print("Test")
+text = "Was ist Tis?"
+
+encoding = tokenizer(text, return_tensors="pt")
+encoding = {k: v.to(trainer.model.device) for k,v in encoding.items()}
+
+outputs = trainer.model(**encoding)
+logits = outputs.logits
+print(logits.shape)
+
+
 print("Save to Space")
 trainer.save_model()
 print("done")