Update app.py

app.py

@@ -31,8 +31,8 @@ def upscale_image(model, input_image):
 # Define the Gradio interface
 iface = gr.Interface(
     fn=upscale_image,
-    inputs=[gr.Radio(["SD 2.0 2x Latent Upscaler", "SD 2.1 4x Upscaler"], label="Models:"), gr.Image(type="filepath")],
-    outputs=gr.Image(type="filepath"),
+    inputs=[gr.Radio(["SD 2.0 2x Latent Upscaler", "SD 2.1 4x Upscaler"], label="Models:"), gr.Image(type="filepath", label = "Raw Image")],
+    outputs=gr.Image(type="filepath", label = "Upscaled Image"),
     title="SD Image Upscaler",
     description="Upscale an image using either the SD 2.0 2x Latent Upscaler or the SD 2.1 4x Upscaler. Use the 4x Upscaler for images lower than 512x512. Use the 2x Upscaler for images 512x512 to 768x768"
 )