Bazyl
commited on
Commit
·
83b41b1
1
Parent(s):
406e919
remove tokenizer
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ from typing import Tuple, Dict
|
|
7 |
|
8 |
def predict(img) -> Tuple[Dict, float]:
|
9 |
start_time = timer()
|
10 |
-
classifier = pipeline("image-classification", model="bazyl/gtsrb-model"
|
11 |
result = classifier(img)
|
12 |
response = {result[i]["label"]: result[i]["score"] for i in range(len(result))}
|
13 |
pred_time = round(timer() - start_time, 5)
|
|
|
7 |
|
8 |
def predict(img) -> Tuple[Dict, float]:
|
9 |
start_time = timer()
|
10 |
+
classifier = pipeline("image-classification", model="bazyl/gtsrb-model")
|
11 |
result = classifier(img)
|
12 |
response = {result[i]["label"]: result[i]["score"] for i in range(len(result))}
|
13 |
pred_time = round(timer() - start_time, 5)
|