Kvikontent commited on
Commit
bd74121
·
1 Parent(s): f437991

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +19 -0
app.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import requests
3
+ from PIL import Image
4
+ from io import BytesIO
5
+
6
+ st.title('CreativeAI')
7
+
8
+ prompt = st.text_input("Enter prompt")
9
+
10
+ if prompt:
11
+ url = 'https://ee1f-217-71-237-228.ngrok-free.app/generate_image'
12
+ params = {'prompt': prompt}
13
+ response = requests.get(url, params=params)
14
+
15
+ if response.status_code == 200:
16
+ image = Image.open(BytesIO(response.content))
17
+ st.image(image, caption='Generated Image', use_column_width=True)
18
+ else:
19
+ st.error("Failed to generate the image. Please try again.")