Update README.md
Browse files
README.md
CHANGED
@@ -65,7 +65,7 @@ from PIL import Image
|
|
65 |
import torch
|
66 |
|
67 |
# Load model and processor
|
68 |
-
model_name = "
|
69 |
model = ViTForImageClassification.from_pretrained(model_name)
|
70 |
processor = ViTImageProcessor.from_pretrained(model_name)
|
71 |
|
|
|
65 |
import torch
|
66 |
|
67 |
# Load model and processor
|
68 |
+
model_name = "prithivMLmods/deepfake-detector-model"
|
69 |
model = ViTForImageClassification.from_pretrained(model_name)
|
70 |
processor = ViTImageProcessor.from_pretrained(model_name)
|
71 |
|