Update app.py

app.py

@@ -5,15 +5,11 @@ from transparent_background import Remover
 import numpy as np
 
 # Initialize the model globally
-remover = Remover()
+remover = Remover(jit=False)
 
 @spaces.GPU
-def process_image(input_image, torchscript_jit, output_type):
+def process_image(input_image, output_type):
     global remover
-    if torchscript_jit == "on" and not isinstance(remover, Remover(jit=True).__class__):
-        remover = Remover(jit=True)
-    elif torchscript_jit == "default" and not isinstance(remover, Remover().__class__):
-        remover = Remover()
     
     if output_type == "Mask only":
         # Process the image and get only the mask
@@ -34,7 +30,6 @@ iface = gr.Interface(
     fn=process_image,
     inputs=[
         gr.Image(type="pil", label="Input Image"),
-        gr.Radio(["default", "on"], label="TorchScript JIT", value="on"),
         gr.Radio(["Default", "Mask only"], label="Output Type", value="Default")
     ],
     outputs=gr.Image(type="pil", label="Output Image"),