vaniv commited on
Commit
7942ef0
·
verified ·
1 Parent(s): 9ef7617

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -49,7 +49,6 @@ def get_outfit_recommendation(pred_class):
49
 
50
  def predict(image):
51
  pred_class, _, _ = learn.predict(image)
52
- pred_class = class_names[pred_idx] # Convert index to class name
53
  dominant_color = get_dominant_color(image)
54
  complementary_color = get_complementary_color(dominant_color)
55
  garment_recommendation = get_outfit_recommendation(pred_class)
 
49
 
50
  def predict(image):
51
  pred_class, _, _ = learn.predict(image)
 
52
  dominant_color = get_dominant_color(image)
53
  complementary_color = get_complementary_color(dominant_color)
54
  garment_recommendation = get_outfit_recommendation(pred_class)