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