Update pages/02_🎥_Input_Youtube_Link.py

pages/02_🎥_Input_Youtube_Link.py

@@ -1,34 +1,16 @@
 import streamlit as st
 from streamlit_lottie import st_lottie
-from models.deep_colorization.colorizers import *
-import requests
-
-st.set_page_config(page_title="Image & Video Colorizer", page_icon="🎨", layout="wide")
 
+from models.deep_colorization.colorizers import eccv16
+from utils import load_lottieurl, change_model
 
-# Define a function that we can use to load lottie files from a link.
-def load_lottieurl(url: str):
-    r = requests.get(url)
-    if r.status_code != 200:
-        return None
-    return r.json()
+st.set_page_config(page_title="Image & Video Colorizer", page_icon="🎨", layout="wide")
 
 
 loaded_model = eccv16(pretrained=True).eval()
 current_model = "None"
 
 
-def change_model(current_model, model):
-    if current_model != model:
-        if model == "ECCV16":
-            loaded_model = eccv16(pretrained=True).eval()
-        elif model == "SIGGRAPH17":
-            loaded_model = siggraph17(pretrained=True).eval()
-        return loaded_model
-    else:
-        raise Exception("Model is the same as the current one.")
-
-
 col1, col2 = st.columns([1, 3])
 with col1:
     lottie = load_lottieurl("https://assets5.lottiefiles.com/packages/lf20_RHdEuzVfEL.json")
@@ -51,7 +33,7 @@ def main():
 
     link = st.text_input("YouTube Link (The longer the video, the longer the processing time)")
     if st.button("Colorize"):
-        print("yo")
+        st.info('This feature hasn\'t been implemented yet', icon="ℹ️")
 
 
 if __name__ == "__main__":