Spaces:
Build error
Build error
Update app.py
Browse files
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,
|
|
|
|
|
|
|
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(
|