Spaces:
Runtime error
Runtime error
File size: 3,794 Bytes
67c45b7 31670dd 67c45b7 31670dd 67c45b7 31670dd 67c45b7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 |
import streamlit as st
from streamlit_drawable_canvas import st_canvas
from PIL import Image
from typing import Union
import random
import numpy as np
import os
import time
st.set_page_config(layout="wide")
def create_edit_existing_image_tab():
st.write("# Edit existing image")
cols = st.columns(2)
with cols[0]:
image_source = st.file_uploader("Upload source image", type=["png", "jpg", "jpeg", "webp"], key="upload_source_edit_existing_image")
st.text_input("Source object", key="text_input_source_edit_existing_image")
st.image('content/dog.png')
with cols[1]:
image_target = st.file_uploader("Upload target image", type=["png", "jpg", "jpeg", "webp"], key="upload_target_edit_existing_image")
st.text_input("Target object", key="text_input_target_edit_existing_image")
st.image('content/cat-sofa.png')
st.text_input("Prompt", key="text_input_prompt_edit_existing_image")
st.text_input("Negative prompt", key="text_input_negative_prompt_edit_existing_image")
st.button("Generate", key="button_generate_edit_existing_image")
st.write("## Result")
st.image('content/after_editing.png')
def create_edit_generated_image_tab():
st.write("# Edit generated image")
cols = st.columns(2)
with cols[0]:
image_source = st.file_uploader("Upload source image", type=["png", "jpg", "jpeg", "webp"], key="upload_source_edit_generated_image")
st.text_input("Target object", key="text_input_source_edit_generated_image")
st.text_input("Prompt", key="text_input_prompt_edit_generated_image")
st.text_input("Negative prompt", key="text_input_negative_prompt_edit_generated_image")
if image_source:
st.button("Generate", key="button_generate_edit_generated_image")
with cols[1]:
st.image('content/dog.png')
st.write("## Result")
cols_result = st.columns(2)
with cols_result[0]:
st.write("### Generated image before editing")
st.image('content/before_editing_generated.png')
with cols_result[1]:
st.write("### Generated image after editing")
st.image('content/after_editing_generated.png')
def create_zero_shot_generation_tab():
st.write("# Zero-shot generation")
def create_zero_shot_stylization_tab():
st.write("# Zero-shot stylization")
def create_home_tab():
st.write("# Home of BLIP-Diffusion")
st.write("Welcome to the demo application of BLIP-Diffusion")
st.write("Project page is [here](https://dxli94.github.io/BLIP-Diffusion-website/.)")
st.write("Github page is [here](https://github.com/salesforce/LAVIS/tree/main/projects/blip-diffusion)")
st.write("Paper is [here](https://arxiv.org/abs/2305.14720)")
st.image('content/teaser-website.png')
def main():
with st.sidebar:
st.title("Navigation")
st.slider("Guidance scale", 0.0, 20.0, 7.5, 0.1)
st.slider("Inference steps", 5, 40, 20, 1)
st.number_input("Seed", 0, 100000, 0, 1)
tab_names = ["Home", "Edit existing image", "Edit generated image", "Zero-shot generation", "Zero-shot stylization"]
(home_tab,
edit_existing_image_tab,
edit_generated_image_tab,
zero_shot_generation_tab,
zero_shot_stylization_tab) = st.tabs(tab_names)
with home_tab:
create_home_tab()
with edit_existing_image_tab:
create_edit_existing_image_tab()
with edit_generated_image_tab:
create_edit_generated_image_tab()
with zero_shot_generation_tab:
create_zero_shot_generation_tab()
with zero_shot_stylization_tab:
create_zero_shot_stylization_tab()
if __name__ == "__main__":
main()
|