Spaces:
Sleeping
Sleeping
Update main.py
Browse files
main.py
CHANGED
@@ -42,7 +42,7 @@ if __name__ == "__main__":
|
|
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),
|
|
|
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),
|