Owos commited on
Commit
7fdea0b
·
1 Parent(s): 1de801f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -40,7 +40,7 @@ if uploaded_file is not None:
40
  col1, col2 = st.columns( [0.5, 0.5])
41
  with col1:
42
  st.markdown('<p style="text-align: center;">Before</p>',unsafe_allow_html=True)
43
- st.image(image,width=500)
44
 
45
  with col2:
46
  st.markdown('<p style="text-align: center;">After</p>',unsafe_allow_html=True)
@@ -70,7 +70,7 @@ if uploaded_file is not None:
70
  with col2:
71
  if 'submitted' in st.session_state:
72
  result = Image.open('out.png')
73
- st.image(result, width=500)
74
  buf = BytesIO()
75
  result.save(buf, format="png")
76
  if len(os.listdir('generated_samples')) <= 10:
@@ -79,8 +79,8 @@ if uploaded_file is not None:
79
  _ = upload_file(path_or_fileobj = 'out.png',
80
  path_in_repo ="remote/" + img_file_name,
81
  repo_id='AfrodreamsAI/afrodreams',
82
- #repo_type='space',
83
- #token=HF_TOKEN
84
  )
85
 
86
  byte_im = buf.getvalue()
 
40
  col1, col2 = st.columns( [0.5, 0.5])
41
  with col1:
42
  st.markdown('<p style="text-align: center;">Before</p>',unsafe_allow_html=True)
43
+ st.image(image,width=300)
44
 
45
  with col2:
46
  st.markdown('<p style="text-align: center;">After</p>',unsafe_allow_html=True)
 
70
  with col2:
71
  if 'submitted' in st.session_state:
72
  result = Image.open('out.png')
73
+ st.image(result, width=300)
74
  buf = BytesIO()
75
  result.save(buf, format="png")
76
  if len(os.listdir('generated_samples')) <= 10:
 
79
  _ = upload_file(path_or_fileobj = 'out.png',
80
  path_in_repo ="remote/" + img_file_name,
81
  repo_id='AfrodreamsAI/afrodreams',
82
+ repo_type='space',
83
+ token=HF_TOKEN
84
  )
85
 
86
  byte_im = buf.getvalue()