Update app.py
Browse files
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[
|
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
|
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)
|
|
|
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)
|