Update app.py

app.py

@@ -157,7 +157,7 @@ print("Done Eval")
 ############################
 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()}
 
@@ -169,29 +169,26 @@ print(logits.shape)
 
 print("Output:\n" )
 #print(tokenizer.decode(outputs[0], skip_special_tokens=True))
-
+'''
 
 print("Save to Space")
 #trainer.save_model("alexkueck/test-tis-1")
 print("done")
 
+'''
 #####################################
 #Push to Hub
 print("push to hub")
 login(token=os.environ["HF_WRITE_TOKEN"]) 
-#trainer.push_to_hub("alexkueck/test-tis-1", use_auth_token=os.environ["HF_ACCESS_TOKEN"])
-my_repo_url = 'https://huggingface.co/alexkueck/test-tis-1'
-#trainer.push_to_hub(repo_url=my_repo_url, use_auth_token=os.environ["HF_ACCESS_TOKEN"])
-#trainer.push_to_hub(repo_name="tis-1", repo_organization="LIF15")
 trainer.push_to_hub("test-tis-1")
 print("done")
-
+'''
 
 
 ##############################################
 #Testen des fine-tuned Modells
 print("Predict")
-model_neu = "EleutherAI/gpt-neo-1.3B"
+model_neu = "test-tis-1"
 tokenizer_neu,model_neu, device_neu = load_tokenizer_and_model(model_neu, False)
 antwort = predict(model_neu, tokenizer_neu, device_neu, "Was ist Tis?", [["Tis", ""]], top_p=5,
             temperature=0.8,