Rename controlnet_depth_canny_segmentation.py to app.py

controlnet_depth_canny_segmentation.py → app.py

@@ -25,7 +25,7 @@ from transformers import Mask2FormerForUniversalSegmentation, Mask2FormerImagePr
 stable_diffusion_base = "runwayml/stable-diffusion-v1-5"
 
 # Path to your fine-tuned ControlNet Depth model
-finetune_controlnet_depth_path = "/content/drive/MyDrive/Output_ControlNet_Finetune"
+finetune_controlnet_depth_path = "controlnt/"
 
 # Pre-trained ControlNet models for Canny and Segmentation
 # These are standard models from the lllyasviel collection, optimized for these tasks.
@@ -345,7 +345,4 @@ with gr.Blocks() as iface:
 load_diffusion_pipeline_and_controlnets()
 
 if __name__ == "__main__":
-    iface.launch(debug=True, share=True)
-
-from google.colab import drive
-drive.mount('/content/drive')
+    iface.launch(debug=True, share=True)