Isaacgonzales
commited on
Commit
•
fa5693e
1
Parent(s):
cf1edb0
update parameters
Browse files
model.py
CHANGED
@@ -48,7 +48,7 @@ def prediction(image_path):
|
|
48 |
p = model_recog(image).softmax(-1)
|
49 |
pred, p = model_recog.tokenizer.decode(p)
|
50 |
t_stop = process_time()
|
51 |
-
t_process = (t_stop - t_start)/
|
52 |
print(f'{image_path}: {pred[0]}, {t_process} seconds')
|
53 |
|
54 |
return img_vis, pred[0], t_process
|
|
|
48 |
p = model_recog(image).softmax(-1)
|
49 |
pred, p = model_recog.tokenizer.decode(p)
|
50 |
t_stop = process_time()
|
51 |
+
t_process = (t_stop - t_start)/3
|
52 |
print(f'{image_path}: {pred[0]}, {t_process} seconds')
|
53 |
|
54 |
return img_vis, pred[0], t_process
|