tsungtao commited on
Commit
1bd1511
1 Parent(s): 4967dbc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -1
app.py CHANGED
@@ -65,4 +65,10 @@ output_images = image_grid(output_images, num_samples // 4, 4)
65
  output_images.save("tao/image.png")
66
 
67
 
68
- gr.Interface.load("models/tsungtao/controlnet-mlsd-202305011046").launch()
 
 
 
 
 
 
 
65
  output_images.save("tao/image.png")
66
 
67
 
68
+ #gr.Interface.load("models/tsungtao/controlnet-mlsd-202305011046").launch()
69
+ def infer(prompt, negative_prompt, image):
70
+ # implement your inference function here
71
+ return output_image
72
+
73
+ # you need to pass inputs and outputs according to inference function
74
+ gr.Interface(fn = infer, inputs = ["text", "text", "image"], outputs = "image").launch()