NKASG commited on
Commit
9b27e1a
·
verified ·
1 Parent(s): 5c923fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -41,7 +41,10 @@ def classify_image(img):
41
  predicted_class = class_labels[np.argmax(predictions)]
42
  highest_prob_index = probs.argmax()
43
 
44
- return { "Material Type":predicted_class, "Cloth Category":categories[highest_prob_index] }
 
 
 
45
 
46
 
47
  iface = gr.Interface(
 
41
  predicted_class = class_labels[np.argmax(predictions)]
42
  highest_prob_index = probs.argmax()
43
 
44
+ return { "Material Type is":predicted_class,
45
+
46
+ "Cloth Category is ":categories[highest_prob_index]
47
+ }
48
 
49
 
50
  iface = gr.Interface(