Update app.py

app.py

@@ -9,14 +9,8 @@ import cv2
 import numpy as np
 import argparse
 
-# Parse command line arguments
-parser = argparse.ArgumentParser()
-parser.add_argument("--config", type=str, default="config/inference_config.yaml", help="Path to the config file")
-parser.add_argument("--share", action="store_true", help="Share the interface if provided")
-args = parser.parse_args()
-
 # Load configuration and models
-config = OmegaConf.load(args.config)
+config = OmegaConf.load("config/inference_config.yaml")
 sd_pipeline = StableDiffusionInpaintPipeline.from_pretrained(
     "runwayml/stable-diffusion-inpainting", torch_dtype=torch.float32
 )
@@ -27,7 +21,7 @@ clipaway = CLIPAway(
     alpha_clip_path=config.alpha_clip_ckpt_pth, 
     config=config, 
     alpha_clip_id=config.alpha_clip_id, 
-    device=config.device, 
+    device="cuda", 
     num_tokens=4
 )
 
@@ -37,7 +31,7 @@ def dilate_mask(mask, kernel_size=5, iterations=5):
     mask = cv2.dilate(np.array(mask), kernel, iterations=iterations)
     return Image.fromarray(mask)
 
-@spaces.GPU(duration=20)
+@spaces.GPU
 def remove_obj(image, uploaded_mask, seed):
     image_pil = image["image"].resize((512, 512), Image.ANTIALIAS)
     mask = dilate_mask(uploaded_mask)
@@ -98,8 +92,4 @@ with gr.Blocks(theme="gradio/monochrome") as demo:
         outputs=result_image
     )
 
-# Launch the interface without caching
-if args.share:
-    demo.launch(share=True)
-else:
-    demo.launch()
+demo.launch()