adisaljusi commited on
Commit ·
816b68d
1
Parent(s): 09f9cfa
Update model name and format label list for clarity
Browse files
app.py
CHANGED
|
@@ -14,15 +14,23 @@ OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
|
|
| 14 |
openai_client = OpenAI(api_key=OPENAI_API_KEY) if OPENAI_API_KEY else None
|
| 15 |
|
| 16 |
# Load models
|
| 17 |
-
vit_classifier = pipeline("image-classification", model="adisaljusi/
|
| 18 |
clip_detector = pipeline(
|
| 19 |
model="openai/clip-vit-large-patch14",
|
| 20 |
task="zero-shot-image-classification",
|
| 21 |
)
|
| 22 |
|
| 23 |
labels_cifar10 = [
|
| 24 |
-
"airplane",
|
| 25 |
-
"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 26 |
]
|
| 27 |
|
| 28 |
|
|
|
|
| 14 |
openai_client = OpenAI(api_key=OPENAI_API_KEY) if OPENAI_API_KEY else None
|
| 15 |
|
| 16 |
# Load models
|
| 17 |
+
vit_classifier = pipeline("image-classification", model="adisaljusi/cifar10-vit")
|
| 18 |
clip_detector = pipeline(
|
| 19 |
model="openai/clip-vit-large-patch14",
|
| 20 |
task="zero-shot-image-classification",
|
| 21 |
)
|
| 22 |
|
| 23 |
labels_cifar10 = [
|
| 24 |
+
"airplane",
|
| 25 |
+
"automobile",
|
| 26 |
+
"bird",
|
| 27 |
+
"cat",
|
| 28 |
+
"deer",
|
| 29 |
+
"dog",
|
| 30 |
+
"frog",
|
| 31 |
+
"horse",
|
| 32 |
+
"ship",
|
| 33 |
+
"truck",
|
| 34 |
]
|
| 35 |
|
| 36 |
|