Update app.py

app.py

@@ -18,12 +18,12 @@ if 'models_loaded' not in st.session_state:
 # Define image processing and classification functions
 def imageprocessing(image):
     encoding = st.session_state.image_processor(image.convert("RGB"), return_tensors="pt")
+    st.write(encoding)
     return encoding
 
 def topwear(encoding):
     outputs = st.session_state.top_wear_model(**encoding)
     predicted_class_idx = outputs.logits.argmax(-1).item()
-    st.write(st.session_state.top_wear_model.config.id2label[predicted_class_idx])
     return st.session_state.top_wear_model.config.id2label[predicted_class_idx]
 
 def patterns(encoding):