fix img

app.py

@@ -30,6 +30,7 @@ generator = torch.manual_seed(0)
 # inference function takes prompt, negative prompt and image
 def infer(prompt, negative_prompt, image):
     # implement your inference function here
+    
 
     cond_input = conditioning_image_transforms(image)
     
@@ -44,14 +45,12 @@ def infer(prompt, negative_prompt, image):
     return output[0]
 
     # you need to pass inputs and outputs according to inference function
-    gr.Interface(fn = infer, inputs = ["text", "text", "image"], outputs = "image").launch()
-
 title = "Categorical Conditioning Controlnet for One-Shot Image Stylization."
 description = "This is a demo on ControlNet which generates images based on the style of the conditioning input." 
 # you need to pass your examples according to your inputs
 # each inner list is one example, each element in the list corresponding to a component in the `inputs`.
 examples = [["1girl, green hair, sweater, looking at viewer, upper body, beanie, outdoors, watercolor, night, turtleneck", "low quality", "wikipe_cond_1.png"]]
-gr.Interface(fn = infer, inputs = ["text", "text", "image"], outputs = "image",
+gr.Interface(fn = infer, inputs = ["text", "text", gr.Image(type="pil") ], outputs = "image",
                     title = title, description = description, examples = examples, theme='gradio/soft').launch()