spdraptor/Photo_edit

1#Script added by SPDraptor2import spaces3from typing import Optional4import subprocess5subprocess.run('pip install flash-attn --no-build-isolation', env={'FLASH_ATTENTION_SKIP_CUDA_BUILD': "TRUE"}, shell=True)6import torch7print("cuda present = ",torch.cuda.is_available())8import os9import sys10import gradio as gr11from PIL import Image12 13from modules import masking_module14 15 16 17DESCRIPTION = "Welcome to Raptor APIs"18 19css = """20  #output {21    height: 500px; 22    overflow: auto; 23    border: 1px solid #ccc; 24  }25"""26 27with gr.Blocks(css=css) as demo:28    gr.Markdown(DESCRIPTION)29    with gr.Tab(label="OBJ_mask"):30        with gr.Row():31            with gr.Column():32                image = gr.Image(label="Input main Picture")33                image_object = gr.Textbox(label="object name")34                mask_btn = gr.Button(value="createMask")35                36            with gr.Column():37                output_mask = gr.Image(label="mask")38                mask_btn.click(masking_module.masking_process,inputs=[image,image_object],outputs=output_mask,api_name="masking_step")39 40demo.launch()