Update app.py

app.py

@@ -118,9 +118,9 @@ def train_function_no_sweeps(base_model_path):   #, train_dataset, test_dataset)
     base_model = get_peft_model(base_model, peft_config)
 
     # Use the accelerator
-    base_model = accelerator.prepare(base_model)
-    train_dataset = accelerator.prepare(train_dataset)
-    test_dataset = accelerator.prepare(test_dataset)
+    base_model = Accelerator.prepare(base_model)
+    train_dataset = Accelerator.prepare(train_dataset)
+    test_dataset = Accelerator.prepare(test_dataset)
 
     timestamp = datetime.now().strftime('%Y-%m-%d_%H-%M-%S')