radinplaid commited on
Commit
244bd49
·
verified ·
1 Parent(s): e59102e

Update app.py

Browse files

Add notification for model download

Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -90,6 +90,11 @@ def server(input, output, session):
90
  model_path = Path("/code/models") / input.model()
91
 
92
  if not model_path.exists():
 
 
 
 
 
93
  hf_download(
94
  model_name="quickmt/" + input.model(),
95
  output_dir=Path("/code/models") / input.model(),
 
90
  model_path = Path("/code/models") / input.model()
91
 
92
  if not model_path.exists():
93
+ ui.notification_show(
94
+ f"Downloading model {input.model()}...",
95
+ type="message",
96
+ duration=3,
97
+ )
98
  hf_download(
99
  model_name="quickmt/" + input.model(),
100
  output_dir=Path("/code/models") / input.model(),