Spaces:
Runtime error
Runtime error
File size: 1,696 Bytes
732b57d |
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 |
# IMPORT LIBRARIES & FUNCTIONS
# External Libraries
import streamlit as st
# Project Functions
# - Model
from src.model.model import load_model, stylize_content_image
from src.model.utils import use_low_resource_settings, suppress_warnings
# - Data
from src.data.data import upload_image
from src.data.utils import remove_source_images
use_low_resource_settings("Yes")
suppress_warnings("Yes")
st.write("# 🖼️Neural🎨Style🖌️Transfer🖼️")
# LOAD MODEL
with st.spinner("🖌️Loading Model"):
model = load_model()
st.success("🖌️Model Loaded")
content_image_def, style_image_def = st.columns(2)
with content_image_def:
st.write("🖼️ Content Image: Image to style")
with style_image_def:
st.write("🎨 Style Image: Style to transfer to content image")
content_image, style_image = st.columns(2)
# UPLOAD IMAGES
with content_image:
ContentColumnTitle = "## 🖼️ Content Image 🖼️"
ContentImageSelectionPrompt = "Pick a Content image"
content_image_file = upload_image(
ContentColumnTitle, ContentImageSelectionPrompt, "Content", "content"
)
with style_image:
StyleColumnTitle = "## 🎨 Style Image 🎨"
StyleImageSelectionPrompt = "Pick a Style image"
style_image_file = upload_image(
StyleColumnTitle, StyleImageSelectionPrompt, "Style", "style"
)
# STYLIZE CONTENT IMAGE
stylize_image = st.button("🖼️🖌️🎨 Start Neural Style Transfer 🖼️🖌️🎨")
if stylize_image:
final_image = stylize_content_image(model, content_image_file, style_image_file)
st.write("# Styled Image:")
st.image(final_image)
try:
remove_source_images()
except:
pass
|