fawadkhan commited on
Commit
7bd25ac
1 Parent(s): d0a6368

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -1
README.md CHANGED
@@ -27,13 +27,17 @@ During preprocessing:
27
  You can use this model directly with a pipeline for image classification:
28
 
29
  \`\`\`python
 
 
30
  from transformers import pipeline
31
- classifier = pipeline('image-classification', model='your_hf_username/your_model_name')
32
 
33
  # To classify an image:
34
  image_path = 'path_to_your_image.jpg'
35
  prediction = classifier(image_path)
36
  print(prediction)
 
 
37
  \`\`\`
38
 
39
  ## Training Procedure
 
27
  You can use this model directly with a pipeline for image classification:
28
 
29
  \`\`\`python
30
+
31
+
32
  from transformers import pipeline
33
+ classifier = pipeline('image-classification', model='fawadkhan/ViT_FineTuned_on_ImagesOASIS')
34
 
35
  # To classify an image:
36
  image_path = 'path_to_your_image.jpg'
37
  prediction = classifier(image_path)
38
  print(prediction)
39
+
40
+
41
  \`\`\`
42
 
43
  ## Training Procedure