Fiqa commited on
Commit
6833ac1
·
verified ·
1 Parent(s): 70a68a3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -93,7 +93,7 @@ def generate_caption_and_image(image, f, p, d):
93
  max_sequence_length=512,
94
  generator=torch.Generator("cpu").manual_seed(0)
95
  ).images
96
- return image[0], image[1]
97
 
98
 
99
 
@@ -104,13 +104,13 @@ def generate_caption_and_image(image, f, p, d):
104
  # generated_image1 =pipe(prompt).images[0]
105
 
106
  # return generated_image, generated_image1
107
- return None, None
108
  # Gradio UI
109
  iface = gr.Interface(
110
  fn=generate_caption_and_image,
111
  inputs=[gr.Image(type="pil", label="Upload Image"), gr.Radio(fabrics, label="Select Fabric"), gr.Radio(patterns, label="Select Pattern"), gr.Radio(textile_designs, label="Select Textile Design")],
112
 
113
- outputs=[gr.Image(label="Generated Design 1"), gr.Image(label="Generated Design 2")],
114
  live=True
115
  )
116
  iface.launch(share=True)
 
93
  max_sequence_length=512,
94
  generator=torch.Generator("cpu").manual_seed(0)
95
  ).images
96
+ return image[
97
 
98
 
99
 
 
104
  # generated_image1 =pipe(prompt).images[0]
105
 
106
  # return generated_image, generated_image1
107
+ return None
108
  # Gradio UI
109
  iface = gr.Interface(
110
  fn=generate_caption_and_image,
111
  inputs=[gr.Image(type="pil", label="Upload Image"), gr.Radio(fabrics, label="Select Fabric"), gr.Radio(patterns, label="Select Pattern"), gr.Radio(textile_designs, label="Select Textile Design")],
112
 
113
+ outputs=[gr.Image(label="Generated Design 1")],
114
  live=True
115
  )
116
  iface.launch(share=True)