Update app.py

app.py

@@ -46,7 +46,7 @@ def train_with_bert(dataset, model_name="bert-base-uncased"):
         save_total_limit=2,
     )
     
-KILL    trainer = Trainer(
+    trainer = Trainer(
         model=model,
         args=training_args,
         train_dataset=tokenized_dataset["train"],
@@ -150,6 +150,6 @@ elif selected_tab == "Settings":
             "use_torch": use_torch,
             "use_bert": use_bert
         })
-        if username and not os.path.exists(f"./results_{username}"):
-            os.makedirs(f"./results_{username}")
+        if username and not os.path.exists(f"./converted/results_{username}"):
+            os.makedirs(f"./converted/results_{username}")
         st.success("Settings saved!")