Update app.py

app.py

@@ -44,7 +44,7 @@ xxx = os.getenv("TOKEN")
 login(token=xxx)
 #Modelle und Tokenizer
 model_name = "alexkueck/LIFineTuned"
-print(f"{model_name}/LIFineTuned/model/")
+
 #Alternativ mit beliebigen Modellen:
 base_model = "project-baize/baize-v2-7b"  #load_8bit = False (in load_tokenizer_and_model)
 #base_model = "TheBloke/airoboros-13B-HF"  #load_8bit = False (in load_tokenizer_and_model)
@@ -87,9 +87,9 @@ lm_datasets = tokenized_datasets.map(
 ####################################################
 
 #Training Args
-model_name = base_model.split("/")[-1]
+#model_name = base_model.split("/")[-1]
 training_args = TrainingArguments(
-    f"{model_name}/LIFineTuned/model/",
+    f"{model_name}/model/",
     evaluation_strategy = "epoch",
     learning_rate=2e-5,
     weight_decay=0.01,