Update app.py

app.py

@@ -30,7 +30,7 @@ class_labels = [
 
 def classify_image(img):
     pred, idx, probs = learn.predict(img)
-    highest_prob_index = probs.argmax()
+    
     img = Image.fromarray((img * 255).astype(np.uint8))
     img = img.resize((224, 224))
     img_array = tf.keras.preprocessing.image.img_to_array(img)
@@ -39,8 +39,9 @@ def classify_image(img):
     
     predictions = model.predict(img_array)
     predicted_class = class_labels[np.argmax(predictions)]
+    highest_prob_index = probs.argmax()
 
-    return "The cloth belongs to category categories[highest_prob_index] a predicted_class material"
+    return  categories[highest_prob_index] 
 
 
 iface = gr.Interface(