lsquaremaster commited on
Commit
5e9bf97
1 Parent(s): f9753a5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -20,7 +20,11 @@ title = "Neural Style Transfer"
20
  description = "Transform your photos into stunning works of art with our Neural Style Transfer app. Simply upload your content image and style image, and watch as the AI-powered model applies the artistic style of your choice to your photos. Unleash your creativity and turn ordinary images into extraordinary masterpieces!"
21
 
22
  # Define labels for the input images
23
- input_labels = ["Content/Base Image", "Style Image"]
 
 
 
 
24
 
25
  # Example images
26
  examples = [
@@ -30,7 +34,7 @@ examples = [
30
 
31
 
32
  interface = gr.Interface(fn=style_transfer,
33
- inputs=["image", "image"],
34
  outputs=["image"],
35
  input_labels=input_labels,
36
  title=title,
 
20
  description = "Transform your photos into stunning works of art with our Neural Style Transfer app. Simply upload your content image and style image, and watch as the AI-powered model applies the artistic style of your choice to your photos. Unleash your creativity and turn ordinary images into extraordinary masterpieces!"
21
 
22
  # Define labels for the input images
23
+ # input_labels = ["Content/Base Image", "Style Image"]
24
+ inputs = [
25
+ gr.Image(label="Content/Base Image"),
26
+ gr.Image(label="Style Image")
27
+ ]
28
 
29
  # Example images
30
  examples = [
 
34
 
35
 
36
  interface = gr.Interface(fn=style_transfer,
37
+ inputs=inputs,
38
  outputs=["image"],
39
  input_labels=input_labels,
40
  title=title,