Update classification.py

classification.py

@@ -17,7 +17,7 @@ def topwear(encoding, top_wear_model):
         outputs = top_wear_model(**encoding)
     logits = outputs.logits
     predicted_class_idx = logits.argmax(-1).item()
-    st.write(f"Top Wear: {top_wear_model.config.id2label[predicted_class_idx]}")
+    # st.write(f"Top Wear: {top_wear_model.config.id2label[predicted_class_idx]}")
     return top_wear_model.config.id2label[predicted_class_idx]
 
 def patterns(encoding, pattern_model):
@@ -25,7 +25,7 @@ def patterns(encoding, pattern_model):
         outputs = pattern_model(**encoding)
     logits = outputs.logits
     predicted_class_idx = logits.argmax(-1).item()
-    st.write(f"Pattern: {pattern_model.config.id2label[predicted_class_idx]}")
+    # st.write(f"Pattern: {pattern_model.config.id2label[predicted_class_idx]}")
     return pattern_model.config.id2label[predicted_class_idx]
 
 def prints(encoding, print_model):
@@ -33,7 +33,7 @@ def prints(encoding, print_model):
         outputs = print_model(**encoding)
     logits = outputs.logits
     predicted_class_idx = logits.argmax(-1).item()
-    st.write(f"Print: {print_model.config.id2label[predicted_class_idx]}")
+    # st.write(f"Print: {print_model.config.id2label[predicted_class_idx]}")
     return print_model.config.id2label[predicted_class_idx]
 
 def sleevelengths(encoding, sleeve_length_model):
@@ -41,7 +41,7 @@ def sleevelengths(encoding, sleeve_length_model):
         outputs = sleeve_length_model(**encoding)
     logits = outputs.logits
     predicted_class_idx = logits.argmax(-1).item()
-    st.write(f"Sleeve Length: {sleeve_length_model.config.id2label[predicted_class_idx]}")
+    # st.write(f"Sleeve Length: {sleeve_length_model.config.id2label[predicted_class_idx]}")
     return sleeve_length_model.config.id2label[predicted_class_idx]
 
 def imageprocessing(image):