fix added

SkinGPT.py

@@ -359,7 +359,14 @@ class SkinGPTClassifier:
 
     def predict(self, image):
         image = image.convert('RGB')
+        print(f"Original image mode: {image.mode}, size: {image.size}")
+
         image_tensor = self.transform(image).unsqueeze(0).to(self.device)
+
+        print(f"Tensor shape: {image_tensor.shape}")
+        print(f"Tensor min/max: {image_tensor.min().item():.4f}/{image_tensor.max().item():.4f}")
+        print(f"Tensor mean: {image_tensor.mean().item():.4f}")
+
         with torch.no_grad():
             diagnosis = self.model.generate(image_tensor)
 

app.py

@@ -49,7 +49,6 @@ warnings.filterwarnings("ignore")
 
 @st.cache_resource
 def get_classifier():
-    st.cache_resource.clear()
     torch.use_deterministic_algorithms(True)
     classifier = SkinGPTClassifier()
     for module in [classifier.model.vit,