mikecho commited on
Commit
4f42ae7
·
verified ·
1 Parent(s): d32c6fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import streamlit as st
2
  from transformers import pipeline
 
3
 
4
  def main():
5
  st.set_page_config(page_title="Unmasked the Target Customers", page_icon="🦜")
@@ -99,10 +100,10 @@ def main():
99
 
100
  if uploaded_file is not None:
101
  print(uploaded_file)
102
- bytes_data = uploaded_file.getvalue()
103
  st.image(uploaded_file, caption="Processing Image", use_column_width=True)
104
 
105
- pipeline_1_out, person_count = pipeline_1_final(uploaded_file)
106
  pipeline_2_age = pipeline_2_final(pipeline_1_out)
107
  pipeline_3_gender = pipeline_3_final(pipeline_1_out)
108
  pipeline_4_emotion = pipeline_3_final(pipeline_1_out)
 
1
  import streamlit as st
2
  from transformers import pipeline
3
+ from PIL import Image
4
 
5
  def main():
6
  st.set_page_config(page_title="Unmasked the Target Customers", page_icon="🦜")
 
100
 
101
  if uploaded_file is not None:
102
  print(uploaded_file)
103
+ image = Image.open(file_name)
104
  st.image(uploaded_file, caption="Processing Image", use_column_width=True)
105
 
106
+ pipeline_1_out, person_count = pipeline_1_final(image)
107
  pipeline_2_age = pipeline_2_final(pipeline_1_out)
108
  pipeline_3_gender = pipeline_3_final(pipeline_1_out)
109
  pipeline_4_emotion = pipeline_3_final(pipeline_1_out)