Commit
•
c709590
1
Parent(s):
38cbc62
Update requirements.txt
Browse files- requirements.txt +11 -49
requirements.txt
CHANGED
@@ -1,49 +1,11 @@
|
|
1 |
-
|
2 |
-
|
3 |
-
|
4 |
-
|
5 |
-
|
6 |
-
|
7 |
-
|
8 |
-
|
9 |
-
|
10 |
-
|
11 |
-
|
12 |
-
|
13 |
-
# Upload images
|
14 |
-
st.sidebar.subheader("Upload Images")
|
15 |
-
uploaded_image1 = st.sidebarfile_uploader("Upload Image", type=["jpg", "jpeg", "png"])
|
16 |
-
uploaded_image2 = st.sidebarfile_uploader("Upload Mask", type=["jpg", "jpeg", "png"])
|
17 |
-
|
18 |
-
# Input text
|
19 |
-
st.subheader("Text Prompts")
|
20 |
-
prompt = st.text_input("Text prompt", "")
|
21 |
-
negative_prompt = st.text_input("Negative prompt", "")
|
22 |
-
|
23 |
-
guidance_scale = st.text_input("Guidance scale", 7.5)
|
24 |
-
steps = st.text_input("Steps", 20)
|
25 |
-
strength = st.text_input("Strength", 1)
|
26 |
-
|
27 |
-
btn_submit = st.button("Generate")
|
28 |
-
|
29 |
-
if btn_submit:
|
30 |
-
init_image = Image.open(uploaded_image1).convert("RGB")
|
31 |
-
st.image(init_image, caption="Uploaded Image", use_column_width=True)
|
32 |
-
mask_image = Image.open(uploaded_image2).convert("RGB")
|
33 |
-
st.image(mask_image, caption="Uploaded Mask", use_column_width=True)
|
34 |
-
|
35 |
-
output = pipe(prompt = prompt,
|
36 |
-
negative_prompt=negative_prompt,
|
37 |
-
image=init_image,
|
38 |
-
mask_image=mask_image,
|
39 |
-
guidance_scale=int(guidance_scale),
|
40 |
-
num_inference_steps=int(steps),
|
41 |
-
strength=strength,
|
42 |
-
generator=generator)
|
43 |
-
|
44 |
-
st.text("Generated Image:")
|
45 |
-
st.image(output.images[0])
|
46 |
-
|
47 |
-
|
48 |
-
if __name__ == "__main__":
|
49 |
-
main()
|
|
|
1 |
+
diffusers==0.22.1
|
2 |
+
matplotlib
|
3 |
+
seaborn
|
4 |
+
Pillow
|
5 |
+
opencv-python
|
6 |
+
scikit-image
|
7 |
+
transformers
|
8 |
+
pandas
|
9 |
+
numpy
|
10 |
+
torch
|
11 |
+
streamlit
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|