prithivMLmods commited on
Commit
92f0b39
·
verified ·
1 Parent(s): af93d0e

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -54,7 +54,7 @@ The model is based on the **Vision Transformer (ViT)**, which treats images as s
54
  from transformers import pipeline
55
 
56
  # Load the model
57
- pipe = pipeline('image-classification', model="Deep-Fake-Detector-Model", device=0)
58
 
59
  # Predict on an image
60
  result = pipe("path_to_image.jpg")
 
54
  from transformers import pipeline
55
 
56
  # Load the model
57
+ pipe = pipeline('image-classification', model="prithivMLmods/Deep-Fake-Detector-Model", device=0)
58
 
59
  # Predict on an image
60
  result = pipe("path_to_image.jpg")