shredder-31 commited on
Commit
e681cc9
·
verified ·
1 Parent(s): 78dda97

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -42,7 +42,7 @@ if __name__ == "__main__":
42
  [img2_path, 5],
43
  ]
44
 
45
- model, vocab, transforms = initialize(root_path)
46
 
47
  interface = gr.Interface(
48
  fn=lambda img, beam_width: ImgCap_inference(img, beam_width, model, vocab, transforms),
 
42
  [img2_path, 5],
43
  ]
44
 
45
+ model, vocab, transforms = initialize()
46
 
47
  interface = gr.Interface(
48
  fn=lambda img, beam_width: ImgCap_inference(img, beam_width, model, vocab, transforms),