Update app.py

app.py

@@ -8,7 +8,7 @@ import numpy as np
 from  PIL import Image, ImageEnhance
 from io import BytesIO
 import streamlit_ext as ste #for download button not to rerun
-import huggingface_hub
+from huggingface_hub import upload_file as 
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
@@ -40,7 +40,7 @@ if uploaded_file is not None:
     col1, col2 = st.columns( [0.5, 0.5])
     with col1:
         st.markdown('<p style="text-align: center;">Before</p>',unsafe_allow_html=True)
-        st.image(image,width=300)  
+        st.image(image,width=500)  
 
     with col2:
         st.markdown('<p style="text-align: center;">After</p>',unsafe_allow_html=True)
@@ -76,7 +76,7 @@ if uploaded_file is not None:
                 if len(os.listdir('generated_samples')) <= 10:
                     img_file_name = f"generated_samples/{str(len(os.listdir('generated_samples')))}.png"
                 
-                    _ = huggingface_hub.upload_file(path_or_fileobj = 'out.png',
+                    _ = upload_file(path_or_fileobj = 'out.png',
                             path_in_repo ="remote/" + img_file_name,
                             repo_id='AfrodreamsAI/afrodreams',
                             repo_type='space',