omerXfaruq commited on
Commit
33de0ba
1 Parent(s): d3d5d4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -0
app.py CHANGED
@@ -8,6 +8,19 @@ index = Index.from_env()
8
 
9
  resize_transform = Resize((250,250))
10
 
 
 
 
 
 
 
 
 
 
 
 
 
 
11
  with gr.Blocks() as demo:
12
  gr.Markdown(
13
  """
 
8
 
9
  resize_transform = Resize((250,250))
10
 
11
+
12
+ from transformers import AutoFeatureExtractor, AutoModel
13
+
14
+ model_ckpt = "google/vit-base-patch16-224-in21k"
15
+ extractor = AutoFeatureExtractor.from_pretrained(model_ckpt)
16
+ model = AutoModel.from_pretrained(model_ckpt)
17
+ hidden_dim = model.config.hidden_size
18
+
19
+ from datasets import load_dataset
20
+
21
+ dataset = load_dataset("HengJi/human_faces")
22
+
23
+
24
  with gr.Blocks() as demo:
25
  gr.Markdown(
26
  """