DiGuaQiu commited on
Commit
96a9c3f
1 Parent(s): 80992c3

Update app.py

Browse files

Add path to weights as environmental variable

Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,5 +1,5 @@
1
  import spaces
2
-
3
  import tempfile
4
  from pathlib import Path
5
 
@@ -141,5 +141,6 @@ with gr.Blocks(css=css, title="MRSegmentator") as iface:
141
 
142
 
143
  if __name__ == "__main__":
 
144
  iface.queue()
145
  iface.launch()
 
1
  import spaces
2
+ import os
3
  import tempfile
4
  from pathlib import Path
5
 
 
141
 
142
 
143
  if __name__ == "__main__":
144
+ os.environ["MRSEG_WEIGHTS_PATH"] = "weights_v1.2"
145
  iface.queue()
146
  iface.launch()