Update README.md

README.md

@@ -41,13 +41,13 @@ from transformers import AutoModelForImageClassification
 import torch
 
 # Load the model from the Hub
-model_name = "YourUsername/vit-stm-artifact-fft" # Replace with your repo name
+model_name = "t0m-R/vit-stm-artifact-fft"
 model = AutoModelForImageClassification.from_pretrained(model_name)
 
 # NOTE: This model requires a custom FFT-based preprocessing function.
 # The 'preprocessed_image' tensor must have a shape of (1, 3, 224, 224).
 # See the "Preprocessing" section for details.
-# preprocessed_image = your_custom_fft_preprocessing_function("path/to/your/stm_image.tiff")
+# preprocessed_image = your_custom_fft_preprocessing_function("path/to/your/stm_image")
 
 # Run inference
 with torch.no_grad():