Update README.md
Browse files
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")
|