Bazyl commited on
Commit
83b41b1
·
1 Parent(s): 406e919

remove tokenizer

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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", tokenizer="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)