Update app.py

app.py

@@ -140,14 +140,14 @@ batch_size = 2
 
 training_args = TrainingArguments(
     #output_dir="alexkueck/test-tis-1",
-    output_dir="./model",
+    output_dir="model",
     overwrite_output_dir = 'True',
     per_device_train_batch_size=batch_size,  #batch_size = 2 for full training
     per_device_eval_batch_size=batch_size,
     evaluation_strategy = "steps",  #oder  
     logging_strategy="steps",   #oder epoch
     logging_steps=10,
-    logging_dir='./logs',
+    logging_dir='logs',
     learning_rate=2e-5,
     weight_decay=0.01,
     save_total_limit = 2,
@@ -195,7 +195,7 @@ print("Done Eval")
 print("Save to ???")
 login(token=os.environ["HF_WRITE_TOKEN"]) 
 #trainer.save_model("test-tis-1")
-trainer.save_model("./model")
+trainer.save_model("model")
 print("done")
 
 #####################################
@@ -251,7 +251,7 @@ print("Output:\n" )
 ########################################
 #mit der predict Funktion
 print("Predict")
-antwort = predict(model_neu, tokenizer, device_neu, prompt, [["Tis", ""]], top_p=5,
+antwort = predict(model_neu, tokenizer, device_neu, prompt, [['Tis', '']], top_p=5,
             temperature=0.8,
             max_length_tokens=1024,
             max_context_length_tokens=2048,)