Hugging Face's logo

Spaces:
Dana19
/
outfit_color_guide
Runtime error

App Files Community
ttcielott commited on
Commit
cf15b33
•
1 Parent(s): fe3defd

revise output image code.

Browse files
Files changed (3) hide show
  1. .DS_Store +0 -0
  2. app.py +3 -1
  3. red outfit_second_color.png → red_sugg.png +0 -0
.DS_Store ADDED
Binary file (6.15 kB). View file
 
app.py CHANGED
@@ -7,7 +7,9 @@ def predict(img):
7
  img = PILImage.create(img)
8
  pred, pred_idx, probs = learn.predict(img)
9
  highest_prob = probs.argmax()
10
- return f'Is this {labels[highest_prob]}?', f'{labels[highest_prob]}_second_color.png'
 
 
11
 
12
  picture = st.camera_input("Take a picture")
13
 
 
7
  img = PILImage.create(img)
8
  pred, pred_idx, probs = learn.predict(img)
9
  highest_prob = probs.argmax()
10
+ file_name = f'{labels[highest_prob].split(' ')[0]}_sugg.png'
11
+ img_sugg = PILImage.create(
12
+ return f'Is this {labels[highest_prob]}?', img_sugg
13
 
14
  picture = st.camera_input("Take a picture")
15
 
red outfit_second_color.png → red_sugg.png RENAMED
File without changes