Update app.py

app.py

@@ -18,8 +18,6 @@ tokenizer       = GPT2TokenizerFast.from_pretrained("nlpconnect/vit-gpt2-image-c
 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))
-    plt.show()
 
     if greedy:
         generated_ids  = model.generate(pixel_values, max_new_tokens = 30)