update

app.py

@@ -20,8 +20,8 @@ import gradio as gr
 import torch
 
 from inference import InferencePipeline
-from trainer import Trainer
-from uploader import upload
+# from trainer import Trainer
+# from uploader import upload
 
 TITLE = '# Custom Diffusion + StableDiffusion Training UI'
 DESCRIPTION = '''This is a demo for [https://github.com/adobe-research/custom-diffusion](https://github.com/adobe-research/custom-diffusion).
@@ -37,7 +37,7 @@ This also reduces the extra storage for each additional concept to 75MB. Our met
 </center>
 '''
 
-ORIGINAL_SPACE_ID = 'nupurkmr9/custom-diffusion'
+ORIGINAL_SPACE_ID = 'Ziqi/ReVersion'
 SPACE_ID = os.getenv('SPACE_ID', ORIGINAL_SPACE_ID)
 SHARED_UI_WARNING = f'''# Attention - This Space doesn't work in this shared UI. You can duplicate and use it with a paid private T4 GPU.
 <center><a class="duplicate-button" style="display:inline-block" target="_blank" href="https://huggingface.co/spaces/{SPACE_ID}?duplicate=true"><img src="https://img.shields.io/badge/-Duplicate%20Space-blue?labelColor=white&style=flat&logo=data:image/png;base64,iVBORw0KGgoAAAANSUhEUgAAABAAAAAQCAYAAAAf8/9hAAAAAXNSR0IArs4c6QAAAP5JREFUOE+lk7FqAkEURY+ltunEgFXS2sZGIbXfEPdLlnxJyDdYB62sbbUKpLbVNhyYFzbrrA74YJlh9r079973psed0cvUD4A+4HoCjsA85X0Dfn/RBLBgBDxnQPfAEJgBY+A9gALA4tcbamSzS4xq4FOQAJgCDwV2CPKV8tZAJcAjMMkUe1vX+U+SMhfAJEHasQIWmXNN3abzDwHUrgcRGmYcgKe0bxrblHEB4E/pndMazNpSZGcsZdBlYJcEL9Afo75molJyM2FxmPgmgPqlWNLGfwZGG6UiyEvLzHYDmoPkDDiNm9JR9uboiONcBXrpY1qmgs21x1QwyZcpvxt9NS09PlsPAAAAAElFTkSuQmCC&logoWidth=14" alt="Duplicate Space"></a></center>
@@ -54,8 +54,8 @@ You can assign a GPU in the {SETTINGS} tab if you are running this on HF Spaces.
 </center>
 '''
 
-os.system("git clone https://github.com/adobe-research/custom-diffusion")
-sys.path.append("custom-diffusion")
+os.system("git clone https://github.com/ziqihuangg/ReVersion")
+sys.path.append("ReVersion")
 
 def show_warning(warning_text: str) -> gr.Blocks:
     with gr.Blocks() as demo: