Update app.py

app.py

@@ -15,6 +15,7 @@ import pandas as pd
 import sklearn
 from sklearn.model_selection import train_test_split
 from sklearn.metrics import accuracy_score, recall_score, precision_score, f1_score
+import pprint
 
 
 #####################################################
@@ -83,7 +84,8 @@ login(token=os.environ["HF_ACCESS_TOKEN"])   #for read access!!!!
 #base_model = "TheBloke/airoboros-13B-HF"  #load_8bit = False (in load_tokenizer_and_model)
 #base_model = "EleutherAI/gpt-neo-1.3B"    #load_8bit = False (in load_tokenizer_and_model)
 #base_model = "TheBloke/airoboros-13B-HF"   #load_8bit = True
-base_model = "TheBloke/vicuna-13B-1.1-HF"   #load_8bit = ?
+#base_model = "TheBloke/vicuna-13B-1.1-HF"   #load_8bit = ?
+base_model="gpt2-large"
 
 #Tokenizer und Model laden
 tokenizer,model,device = load_tokenizer_and_model(base_model,  False)