Update app.py

app.py

@@ -16,7 +16,6 @@ from pipeline_flux_control_removal import FluxControlRemovalPipeline
 
 torch.set_grad_enabled(False)
 os.environ['GRADIO_TEMP_DIR'] = './tmp'
-os.environ["flux_train"]
 image_path = mask_path = None
 image_examples = [...] 
 image_path = mask_path =None
@@ -54,7 +53,7 @@ image_examples = [
 
 def load_model(base_model_path, lora_path):
     global pipe
-    transformer = FluxTransformer2DModel.from_pretrained(base_model_path, use_auth_token=os.getenv("flux_train"), subfolder='transformer', torch_dtype=torch.bfloat16)
+    transformer = FluxTransformer2DModel.from_pretrained(base_model_path, subfolder='transformer', torch_dtype=torch.bfloat16)
     gr.Info(str(f"Model loading: {int((40 / 100) * 100)}%"))
     # enable image inputs
     with torch.no_grad():