snehalsapkale commited on
Commit
887334b
·
verified ·
1 Parent(s): cf2df42

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -6,8 +6,7 @@ import requests
6
 
7
  img_file_buffer = st.file_uploader('Upload a PNG image', type='png')
8
  if img_file_buffer is not None:
9
- image = Image.open(img_file_buffer)
10
- img_array = np.array(image)
11
 
12
  feature_extractor = ViTFeatureExtractor.from_pretrained('rizvandwiki/gender-classification')
13
  model = ViTForImageClassification.from_pretrained('rizvandwiki/gender-classification')
 
6
 
7
  img_file_buffer = st.file_uploader('Upload a PNG image', type='png')
8
  if img_file_buffer is not None:
9
+ image = Image.open(img_file_buffer)
 
10
 
11
  feature_extractor = ViTFeatureExtractor.from_pretrained('rizvandwiki/gender-classification')
12
  model = ViTForImageClassification.from_pretrained('rizvandwiki/gender-classification')