Update gradio_canny2image.py

GaParmar · Mar 20, 2024 · 7d4ebb4 · 7d4ebb4
1 parent 69b05b4
commit 7d4ebb4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gradio_canny2image.py b/gradio_canny2image.py
@@ -6,7 +6,7 @@
 from src.image_prep import canny_from_pil
 from src.pix2pix_turbo import Pix2Pix_Turbo
 
-model = Pix2Pix_Turbo("canny_to_image")
+model = Pix2Pix_Turbo("edge_to_image")
 
 
 def process(input_image, prompt, low_threshold, high_threshold):