Spaces:
Sleeping
Sleeping
AjithBharadwaj
commited on
Commit
•
5109866
1
Parent(s):
f659518
Update app.py
Browse files
app.py
CHANGED
@@ -26,10 +26,10 @@ def main():
|
|
26 |
image_bytes = query({
|
27 |
"inputs": text_input,
|
28 |
})
|
29 |
-
|
30 |
-
|
31 |
-
|
32 |
-
|
33 |
|
34 |
if __name__ == "__main__":
|
35 |
main()
|
|
|
26 |
image_bytes = query({
|
27 |
"inputs": text_input,
|
28 |
})
|
29 |
+
image = Image.open(io.BytesIO(image_bytes))
|
30 |
+
|
31 |
+
# Display the generated image
|
32 |
+
st.image(image, caption='Generated Image', use_column_width=True)
|
33 |
|
34 |
if __name__ == "__main__":
|
35 |
main()
|