Update app.py

app.py

@@ -14,7 +14,7 @@ model_trained.load_state_dict(torch.load('model_trained.pth',map_location=torch.
 image_processor = ViTImageProcessor.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
 tokenizer       = GPT2TokenizerFast.from_pretrained("nlpconnect/vit-gpt2-image-captioning")
 
-def show_n_generate(img, greedy = True, model):
+def show_n_generate(img, model, greedy = True):
     image = Image.open(img)
     pixel_values   = image_processor(image, return_tensors ="pt").pixel_values
     plt.imshow(np.asarray(image))