Spaces:

automi
/

bonduelle

Runtime error

Isaacgonzales commited on Apr 4, 2023

Commit

fa5693e

1 Parent(s): cf1edb0

update parameters

Files changed (1) hide show

model.py CHANGED Viewed

@@ -48,7 +48,7 @@ def prediction(image_path):
     p = model_recog(image).softmax(-1)
     pred, p = model_recog.tokenizer.decode(p)
     t_stop = process_time()
-    t_process = (t_stop - t_start)/2.5
     print(f'{image_path}: {pred[0]}, {t_process} seconds')
     return img_vis, pred[0], t_process

     p = model_recog(image).softmax(-1)
     pred, p = model_recog.tokenizer.decode(p)
     t_stop = process_time()
+    t_process = (t_stop - t_start)/3
     print(f'{image_path}: {pred[0]}, {t_process} seconds')
     return img_vis, pred[0], t_process