update mmcv install from app.py -- li

This commit is contained in:
memoryunreal
2023-04-26 02:11:25 +00:00
parent 242f69052d
commit 1dcb4d91c7
2 changed files with 6 additions and 3 deletions

6
app.py
View File

@@ -14,6 +14,10 @@ import json
import torchvision
import torch
from tools.painter import mask_painter
try:
from mmcv.cnn import ConvModule
except:
os.system("mim install mmcv")
# download checkpoints
def download_checkpoint(url, folder, filename):
@@ -365,7 +369,7 @@ with gr.Blocks() as iface:
video_input = gr.Video(autosize=True)
with gr.Column():
video_info = gr.Textbox()
video_info = gr.Textbox(value="If you want to use the inpaint function, it is best to download and use a machine with more VRAM locally. \
resize_info = gr.Textbox(value="If you want to use the inpaint function, it is best to download and use a machine with more VRAM locally. \
Alternatively, you can use the resize ratio slider to scale down the original image to around 360P resolution for faster processing.")
resize_ratio_slider = gr.Slider(minimum=0.02, maximum=1, step=0.02, value=1, label="Resize ratio", visible=True)