Update app.py

app.py

@@ -1,3 +1,5 @@
+import inspect
+print(inspect.signature(ShapEImg2ImgPipeline.__call__))
 import gradio as gr
 import torch
 import math
@@ -24,7 +26,8 @@ def infer(source_img, prompt, steps, seed, strength):
         steps = math.ceil(1 / max(0.10, strength))
     source_image = resize(512, source_img)
     source_image.save('source.png')
-    result = pipe(prompt=prompt, image=source_image, strength=strength, guidance_scale=7.5, num_inference_steps=steps, generator=generator)
+    # Adjust this line based on the correct method signature
+    result = pipe(images=source_image, strength=strength, guidance_scale=7.5, num_inference_steps=steps, generator=generator)
     return result.images[0]
 
 # Gradio interface
@@ -39,4 +42,4 @@ gr.Interface(
     ],
     outputs='image',
     title="Creative Touch"
-).queue(max_size=10).launch()
+).queue(max_size=10).launch()