Update app.py

app.py

@@ -2,7 +2,6 @@
 
 import gradio as gr
 import tensorflow as tf
-# import os
 import tensorflow_hub as hub
 import PIL
 from PIL import Image,ImageOps
@@ -38,25 +37,9 @@ def tensor_to_image(tensor):
     tensor = tensor[0]
   return PIL.Image.fromarray(tensor)
 
-"""##Saving unscaled Tensor images."""
-
-def save_image(image, filename):
-  """
-    Saves unscaled Tensor Images.
-    Args:
-      image: 3D image tensor. [height, width, channels]
-      filename: Name of the file to save to.
-  """
-  if not isinstance(image, Image.Image):
-    image = tf.clip_by_value(image, 0, 255)
-    image = Image.fromarray(tf.cast(image, tf.uint8).numpy())
-  image.save("%s.jpg" % filename)
-  print("Saved as %s.jpg" % filename)
-
-"""## Grayscaling image for testing purpose to check if we could get better results."""
-
 
 
+"""## Grayscaling image for testing purpose to check if we could get better results."""
 def gray_scaled(inp_img):
   gray = cv2.cvtColor(inp_img, cv2.COLOR_BGR2GRAY)
   gray_img = np.zeros_like(inp_img)
@@ -65,6 +48,7 @@ def gray_scaled(inp_img):
   gray_img[:,:,2] = gray
   return gray_img
 
+##Transformation 
 def transform_mymodel(content_image,style_image):
   # Convert to float32 numpy array, add batch dimension, and normalize to range [0, 1]
   #content_image=gray_scaled(content_image)
@@ -86,8 +70,8 @@ def gradio_intrface(mymodel):
 # Initializing the input component 
   image1 = gr.inputs.Image(label="Content Image") #CONTENT IMAGE
   image2 = gr.inputs.Image(label="Style Image") #STYLE IMAGE
-  stylizedimg=gr.outputs.Image() 
-  gr.Interface(fn=mymodel, inputs= [image1,image2] , outputs= stylizedimg,title='Style Transfer',theme='seafoam',examples=[['Content_Images/contnt12.jpg','VG516.jpg']],article="**References**\n\n""<a href='https://www.tensorflow.org/hub/tutorials/tf2_arbitrary_image_stylization'</a>\n").launch()
+  stylizedimg=gr.outputs.Image(label="Result") 
+  gr.Interface(fn=mymodel, inputs= [image1,image2] , outputs= stylizedimg,title='Style Transfer',theme='seafoam',examples=[['Content_Images/contnt12.jpg','VG516.jpg']],article="References-\n\n""<a href='https://www.tensorflow.org/hub/tutorials/tf2_arbitrary_image_stylization'>\n").launch()
 
 """The function will be launched both  Inline and Outline  where u need to add a content and style image."""