Hugging Face's logo

Spaces:
dj-dawgs-ipd
/
IPD-Image-ViT-Finetune
Running

App Files Community
Tirath5504 commited on
Commit
529e066
•
1 Parent(s): 5e6cfb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import ViTForImageClassification, ViTImageProcessor
3
  from PIL import Image
4
  import gradio as gr
5
 
6
- model = ViTForImageClassification.from_pretrained('vit-hateful-gesture-classification')
7
  processor = ViTImageProcessor.from_pretrained('google/vit-base-patch16-224')
8
 
9
  class_names = ['cut_throat_gesture', 'finger_gun_to_the_head', 'middle_finger', 'slanted_eyes_gesture', 'swastika']
 
3
  from PIL import Image
4
  import gradio as gr
5
 
6
+ model = ViTForImageClassification.from_pretrained('Tirath5504/IPD-Image-ViT-Finetune')
7
  processor = ViTImageProcessor.from_pretrained('google/vit-base-patch16-224')
8
 
9
  class_names = ['cut_throat_gesture', 'finger_gun_to_the_head', 'middle_finger', 'slanted_eyes_gesture', 'swastika']