added GPU implementation

app.py

@@ -1,8 +1,35 @@
+import os
 import gradio as gr
 from video_watermark_remover import *
 from video_converter import *
 from image_converter import *
 from image_watermark_remover import *
+from typing import List
+from pydantic import BaseModel
+from lama_cleaner.server import main
+
+class FakeLamaArgs(BaseModel):
+    host: str = "0.0.0.0"
+    port: int = 5000
+    model: str = 'lama'
+    hf_access_token: str = ""
+    sd_disable_nsfw: bool = False
+    sd_cpu_textencoder: bool = True
+    sd_run_local: bool = False
+    sd_enable_xformers: bool = False
+    local_files_only: bool = False
+    cpu_offload: bool = False
+    device: str = "cuda"
+    gui: bool = False
+    gui_size: List[int] = [1000, 1000]
+    input: str = ''
+    disable_model_switch: bool = True
+    debug: bool = False
+    no_half: bool = False
+    disable_nsfw: bool = False
+    enable_xformers: bool = True
+    model_dir: str = None
+    output_dir: str = None
 
 css = """
     #remove_btn {
@@ -111,4 +138,12 @@ with demo:
     
     gr.Markdown("""## <center style="margin:20px;">Developed by Muhammad Ahmed<img src="https://avatars.githubusercontent.com/u/63394104?v=4" style="height:50px;width:50px;border-radius:50%;margin:5px;"></img></center>
     """)
+
+# Change the code according to the error
+import threading
+
+thread = threading.Thread(target=main, kwargs={'args': FakeLamaArgs()})
+thread.daemon = True
+thread.start()
+
 demo.launch(show_api=False)

image_watermark_remover.py

@@ -52,7 +52,7 @@ def remove_image_watermark(input):
         'mask': ('mask.jpg', mask_data),
     }
 
-    response = requests.post('https://ahmedghani-lama-cleaner-lama.hf.space/inpaint', data=form_data, files=files_data)
+    response = requests.post('http://localhost:5000/inpaint', data=form_data, files=files_data)
 
     if response.headers['Content-Type'] == 'image/jpeg' or response.headers['Content-Type'] == 'image/png':
         image = Image.open(io.BytesIO(response.content))

requirements.txt

@@ -3,4 +3,6 @@ ffmpeg-python
 moviepy
 pydub
 opencv-python
-pyheif
+pyheif
+lama-cleaner==0.33.0
+xformers==0.0.16

video_watermark_remover.py

@@ -55,7 +55,7 @@ def process_image(mask_data, image_path):
         'mask': ('mask.png', mask_data)
     }
 
-    response = requests.post('https://ahmedghani-lama-cleaner-lama.hf.space/inpaint', data=form_data, files=files_data)
+    response = requests.post('http://localhost:5000/inpaint', data=form_data, files=files_data)
 
     if response.headers['Content-Type'] == 'image/jpeg' or response.headers['Content-Type'] == 'image/png':
         output_image_path = os.path.join('output_images', os.path.splitext(os.path.basename(image_path))[0] + '_inpainted' + os.path.splitext(image_path)[1])