Tirath5504 commited on
Commit
b594dbc
·
verified ·
1 Parent(s): be8f613

Update app.py

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