Update app.py

app.py

@@ -88,8 +88,8 @@ def process(
        seed_slicer = random.randint(0, MAX_SEED)
     generator = torch.Generator().manual_seed(seed_slicer)
  
-    if not mask:
-        gr.Info("Please draw a mask on the image.")
+    # if not mask:
+       # gr.Info("Please draw a mask on the image.")
         pipe2.load_lora_weights("SIGMitch/KIT")
         result = pipe2(
         prompt=input_text,
@@ -102,9 +102,9 @@ def process(
         num_inference_steps=num_inference_steps_slider
         ).images[0]
         print('INFERENCE DONE')
-        return result, None
+    return result, None
 
-    resized_mask = mask.resize((width, height), Image.LANCZOS)
+     #resized_mask = mask.resize((width, height), Image.LANCZOS)
 
    #pipe.load_lora_weights("SIGMitch/KIT")
     #result = pipe(
@@ -119,7 +119,7 @@ def process(
      #   num_inference_steps=num_inference_steps_slider
     #).images[0]
     #print('INFERENCE DONE')
-    return result, resized_mask
+    # return result, resized_mask
 
 
 with gr.Blocks() as demo: