Spaces:
Runtime error
Runtime error
omerXfaruq
commited on
Commit
•
33de0ba
1
Parent(s):
d3d5d4c
Update app.py
Browse files
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 |
"""
|