NKASG commited on
Commit
090e57f
·
verified ·
1 Parent(s): 520fe8b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def classify_image(img):
40
  predictions = model.predict(img_array)
41
  predicted_class = class_labels[np.argmax(predictions)]
42
 
43
- return The cloth belongs to category categories[highest_prob_index] a predicted_class material
44
 
45
 
46
  iface = gr.Interface(
 
40
  predictions = model.predict(img_array)
41
  predicted_class = class_labels[np.argmax(predictions)]
42
 
43
+ return "The cloth belongs to category categories[highest_prob_index] a predicted_class material"
44
 
45
 
46
  iface = gr.Interface(