kubinooo commited on
Commit
b57ddbf
·
verified ·
1 Parent(s): 0c20a4d

update usage

Browse files
Files changed (1) hide show
  1. README.md +7 -5
README.md CHANGED
@@ -100,11 +100,13 @@ import torch
100
  processor = AutoImageProcessor.from_pretrained("kubinooo/convnext-tiny-224-audio-deepfake-classification")
101
  model = AutoModelForImageClassification.from_pretrained("kubinooo/convnext-tiny-224-audio-deepfake-classification")
102
 
103
- image = Image.open("mel_spectrogram_image.png")
104
- inputs = processor(images=image, return_tensors="pt")
105
- outputs = model(**inputs)
106
- logits = outputs.logits
107
- predicted_class = logits.argmax(-1).item()
 
 
108
  ```
109
 
110
  ### Downstream Use
 
100
  processor = AutoImageProcessor.from_pretrained("kubinooo/convnext-tiny-224-audio-deepfake-classification")
101
  model = AutoModelForImageClassification.from_pretrained("kubinooo/convnext-tiny-224-audio-deepfake-classification")
102
 
103
+ image = Image.open("/path/to/your/image").convert("RGB")
104
+ pixel_values = processor(image, return_tensors="pt").pixel_values
105
+ with torch.no_grad():
106
+ outputs = model(pixel_values)
107
+ logits = outputs.logits
108
+ predicted_class_idx = logits.argmax(-1).item()
109
+ print(model.config.id2label[predicted_class_idx])
110
  ```
111
 
112
  ### Downstream Use