22

app.py

@@ -79,21 +79,21 @@ spaces.aoti_blocks_load(pipe.transformer.layers,
 def prepare(prompt, input_image, control_mode='Canny'):
     polished_prompt = polish_prompt(prompt)
 
-    if not input_image: return polished_prompt
-    
-    if control_mode == 'HED':
-        processor = HEDdetector.from_pretrained("lllyasviel/Annotators")
-    if control_mode =='Midas':
-        processor = MidasDetector.from_pretrained("lllyasviel/Annotators")
-    if control_mode =='MLSD':
-        processor = MLSDdetector.from_pretrained("lllyasviel/Annotators")
-    if control_mode =='Pose':
-        processor = OpenposeDetector.from_pretrained("lllyasviel/Annotators")
-    else:
-        processor = CannyDetector()
-    control_image = processor(input_image)
+    return polished_prompt
+
+    # if control_mode == 'HED':
+    #     processor = HEDdetector.from_pretrained("lllyasviel/Annotators")
+    # if control_mode =='Midas':
+    #     processor = MidasDetector.from_pretrained("lllyasviel/Annotators")
+    # if control_mode =='MLSD':
+    #     processor = MLSDdetector.from_pretrained("lllyasviel/Annotators")
+    # if control_mode =='Pose':
+    #     processor = OpenposeDetector.from_pretrained("lllyasviel/Annotators")
+    # else:
+    #     processor = CannyDetector()
+    # control_image = processor(input_image)
     
-    return polished_prompt, control_image
+    # return polished_prompt, control_image
 
 @spaces.GPU
 def inference(