Spaces:

bnoy1
/

rock-paper-scissors-class

Runtime error

bnoy1 commited on 7 days ago

Commit

42a7263

verified ·

1 Parent(s): d19ec4c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -8,7 +8,7 @@ import os
 dataset = load_dataset("imagefolder", data_dir=".", split={"train": "train[:80%]", "test": "train[80%:]"})
 # בוחרים מודל בסיסי
-checkpoint = "google/vit-tiny-patch16-224"
 processor = AutoImageProcessor.from_pretrained(checkpoint)
 model = AutoModelForImageClassification.from_pretrained(
     checkpoint,

 dataset = load_dataset("imagefolder", data_dir=".", split={"train": "train[:80%]", "test": "train[80%:]"})
 # בוחרים מודל בסיסי
+checkpoint = "facebook/deit-tiny-patch16-224"
 processor = AutoImageProcessor.from_pretrained(checkpoint)
 model = AutoModelForImageClassification.from_pretrained(
     checkpoint,