Update app.py

app.py

@@ -11,7 +11,7 @@ from PIL import Image
 from create_maxim_model import Model
 from maxim.configs import MAXIM_CONFIGS
 
-print(f"Number of GPUs detected by TF: {tf.config.list_physical_devices('GPU')}")
+
 _MODEL = from_pretrained_keras("sayakpaul/S-2_enhancement_lol")
 
 
@@ -97,7 +97,7 @@ description = "The underlying model is [this](https://huggingface.co/sayakpaul/S
 iface = gr.Interface(
     infer,
     inputs="image",
-    outputs="image",
+    outputs=gr.Image().style(height=242),
     title=title,
     description=description,
     allow_flagging="never",