Upload app.py with huggingface_hub

app.py

@@ -1,10 +1,13 @@
 import neural_style
 
 import streamlit as st
+import os 
+import random
 import numpy as np
 #import cv2
 from  PIL import Image, ImageEnhance
 
+st.set_page_config(layout="wide")
 #Create two columns with different width
 col1, col2 = st.columns( [0.8, 0.2])
 with col1:               # To display the header text using css style
@@ -42,7 +45,11 @@ if uploaded_file is not None:
         params.backend = "mkl"
         params.image_size = 128
         params.content_image = uploaded_file
-        neural_style.transfer(params)
+        path = 'style'
+        styles = os.listdir(path)
+        params.style_image = random.choice(styles)
+        with st.spinner('Wait for it...'):
+            neural_style.transfer(params)
 
         #display image when done.
         with col2: