Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from ModelLoader import ModelLoader
+import torch
 import os
 
 max_img_wh = 4096  # Set the maximum image size in pixels
@@ -15,11 +16,11 @@ demo = gr.Interface(
     fn=inference,
     inputs=[
         gr.components.Image(type='pil', label='Input Image'),
-        gr.components.Checkbox(label='Use GPU', value=True)  # Precheck the GPU checkbox
+        gr.components.Checkbox(label='Use GPU', value=torch.cuda.is_available())  # Precheck the GPU checkbox
     ],
     outputs=gr.components.Image(type='pil', label='Output Image', format='jpeg'),
-    title='UnderWater Photo - published by Yarflam',
-    description='Upload an image to see the model output.',
+    title="Underwater Photo Enhancement",
+    description="This tool enhances the quality of underwater photos by improving clarity and color. Upload your underwater photo to see the enhanced version.",
     flagging_mode='never'
 )
 demo.launch()