hysts HF Staff commited on
Commit
79b969d
·
1 Parent(s): d87e974
Files changed (1) hide show
  1. app.py +12 -5
app.py CHANGED
@@ -30,15 +30,22 @@ After training, you can press "Reload Weight List" button to load your trained m
30
  Note that your trained models will be deleted when the second training is started.
31
  '''
32
 
33
- SPACE_ID = os.getenv('SPACE_ID', 'hysts/LoRA-SD-training')
 
34
  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.
35
 
36
  <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>
37
  '''
38
- CUDA_NOT_AVAILABLE_WARNING = '''# Attention - CUDA is not available in this environment.
39
-
40
- You can assign a GPU from the Settings tab if you are running this on HF Spaces.
41
- T4 small is sufficient to run this demo.
 
 
 
 
 
 
42
  '''
43
 
44
 
 
30
  Note that your trained models will be deleted when the second training is started.
31
  '''
32
 
33
+ ORIGINAL_SPACE_ID = 'hysts/LoRA-SD-training'
34
+ SPACE_ID = os.getenv('SPACE_ID', ORIGINAL_SPACE_ID)
35
  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.
36
 
37
  <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>
38
  '''
39
+ if os.getenv('SYSTEM') == 'space' and SPACE_ID != ORIGINAL_SPACE_ID:
40
+ SETTINGS = '<a href="{https://huggingface.co/spaces/{SPACE_ID}/settings}">Settings</a>'
41
+
42
+ else:
43
+ SETTINGS = 'Settings'
44
+ CUDA_NOT_AVAILABLE_WARNING = f'''# Attention - Running on CPU.
45
+ <center>
46
+ You can assign a GPU in the {SETTINGS} tab if you are running this on HF Spaces.
47
+ "T4 small" is sufficient to run this demo.
48
+ </center>
49
  '''
50
 
51