Update app.py

app.py

@@ -3,9 +3,7 @@ import gradio as gr
 from PIL import Image
 from transparent_background import Remover
 import numpy as np
-import torch
 
-torch.jit.script = lambda f: f
 # Initialize the model globally
 remover = Remover(jit=False)
 
@@ -27,6 +25,12 @@ def process_image(input_image, output_type):
         # Process the image and return the RGBA result
         output = remover.process(input_image, type='rgba')
         return output
+        
+description = """<h1 align="center">InSPyReNet Background Remover</h1>
+<p><center>
+<a href="https://github.com/plemeri/InSPyReNet" target="_blank">[Github]</a>
+</center></p>
+"""
 
 iface = gr.Interface(
     fn=process_image,
@@ -35,8 +39,7 @@ iface = gr.Interface(
         gr.Radio(["Default", "Mask only"], label="Output Type", value="Default")
     ],
     outputs=gr.Image(type="pil", label="Output Image"),
-    title="Inspyrenet Background Remover",
-    description="Remove the background from an image using Inspyrenet. Choose 'Mask only' for a black and white mask, or 'Default' for the image with transparent background.",
+    description=description,
     theme='bethecloud/storj_theme',
 )