Update README.md

README.md

@@ -23,26 +23,21 @@ The trained model is accessible on Hugging Face Hub at: [tribber93/my-trash-clas
 
 To use the model for inference, follow these steps:
 
-1. Load the pre-trained model from Hugging Face:
-
 ```python
+import torch
+import requests
+from PIL import Image
 from transformers import AutoModelForImageClassification, AutoImageProcessor
 
+url = 'https://cdn.grid.id/crop/0x0:0x0/700x465/photo/grid/original/127308_kaleng-bekas.jpg'
+image = Image.open(requests.get(url, stream=True).raw)
+
 model_name = "tribber93/my-trash-classification"
 model = AutoModelForImageClassification.from_pretrained(model_name)
-image_processor = AutoImageProcessor.from_pretrained(model_name)
-```
-
-2. Prepare an image for prediction:
-
-```python
-from PIL import Image
-import torch
-
-image = Image.open("path_to_image.jpg").convert("RGB")
-pixel_values = image_processor(image, return_tensors="pt").pixel_values
+processor = AutoImageProcessor.from_pretrained(model_name)
+inputs = processor(image, return_tensors="pt")
 
-outputs = model(pixel_values)
+outputs = model(**inputs)
 predictions = torch.argmax(outputs.logits, dim=-1)
 print("Predicted class:", model.config.id2label[predictions.item()])
 ```