Upload 2 files

app.py

@@ -1,12 +1,9 @@
 import gradio as gr
-import cv2
-from flask import Flask,request,render_template
 import tensorflow as tf
 from PIL import Image
 from tensorflow.keras.applications import vgg19
 
 from tensorflow.python.keras import models 
-import IPython.display
 import numpy as np
 from tensorflow import keras
 from werkzeug.utils import secure_filename
@@ -170,12 +167,7 @@ def run_style_transfer(content_path,style_path,epochs=20,content_weight=1e3, sty
             plot_img = noise.numpy()
             plot_img = deprocess_img(plot_img)
             imgs.append(plot_img)
-            IPython.display.clear_output(wait=True)
-            IPython.display.display_png(Image.fromarray(plot_img))
-        
-    
-    IPython.display.clear_output(wait=True)
-    
+
     
     return best_img,best_loss,imgs