Spaces:

DKatheesrupan
/

Exercise2

Running

DKatheesrupan commited on Apr 21

Commit

bb2ea71

verified ·

1 Parent(s): 4e50c6a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ from transformers import pipeline
 BASE_DIR = Path(__file__).resolve().parent
 EXAMPLE_DIR = BASE_DIR / "example_images"
-MODEL_PATH = "DKatheesrupan/aufgabe2"
 CAT_LABELS = ["cheetah", "leopard", "lion", "puma", "tiger"]

 BASE_DIR = Path(__file__).resolve().parent
 EXAMPLE_DIR = BASE_DIR / "example_images"
+MODEL_PATH = "DKatheesrupan/cat-vit"
 CAT_LABELS = ["cheetah", "leopard", "lion", "puma", "tiger"]