Update app.py

app.py

@@ -50,6 +50,9 @@ import torch
 import re 
 import gradio as gr
 from transformers import AutoTokenizer, ViTFeatureExtractor, VisionEncoderDecoderModel 
+import os
+import tensorflow as tf
+os.environ['TF_ENABLE_ONEDNN_OPTS'] = '0'
 
 device='cpu'
 encoder_checkpoint = "nlpconnect/vit-gpt2-image-captioning"