fix: minor errors

app.py

@@ -1,4 +1,3 @@
-### FINAL COMBINED
 import torch
 from diffusers import StableDiffusionPipeline, StableDiffusionInpaintPipeline
 import os
@@ -17,7 +16,6 @@ import spaces
 choker_images = [Image.open(os.path.join("choker", x)) for x in os.listdir("choker")]
 person_images = [Image.open(os.path.join("without_necklace", x)) for x in os.listdir("without_necklace")]
 
-# initialising the stable diffusion model
 model_id = "stabilityai/stable-diffusion-2-inpainting"
 pipeline = StableDiffusionInpaintPipeline.from_pretrained(
     model_id, torch_dtype=torch.float16
@@ -25,15 +23,13 @@ pipeline = StableDiffusionInpaintPipeline.from_pretrained(
 pipeline = pipeline.to("cuda")
 
 
-# functions
-def clearFunc():
+def clear_func():
     torch.cuda.empty_cache()
     gc.collect()
 
 
-# SD Model
 @spaces.GPU
-def sd_inpaint(image, jewellery):
+def clothing_try_on_n_necklace_try_on(image, jewellery):
     image = np.array(image)
     copy_image = image.copy()
     jewellery = np.array(jewellery)
@@ -154,12 +150,10 @@ def sd_inpaint(image, jewellery):
     results = [
         Image.fromarray(np.bitwise_or(x, np.array(jewellery_mask))) for x in results
     ]
-    clearFunc()
+    clear_func()
     return results[0]
 
 
-# interface
-
 with gr.Blocks() as interface:
     with gr.Row():
         inputImage = gr.Image(label="Input Image", type="pil", image_mode="RGB", interactive=True)
@@ -172,6 +166,6 @@ with gr.Blocks() as interface:
     with gr.Row():
         submit = gr.Button("Enter")
 
-    submit.click(fn=sd_inpaint, inputs=[inputImage, selectedNecklace], outputs=[outputOne, outputTwo, outputThree])
+    submit.click(fn=clothing_try_on_n_necklace_try_on, inputs=[inputImage, selectedNecklace], outputs=[outputOne])
 
 interface.launch(debug=True)