andreped commited on
Commit
c2b6b34
1 Parent(s): dc74665

Fixed task in demo

Browse files
Files changed (2) hide show
  1. demo/app.py +2 -0
  2. demo/src/inference.py +1 -1
demo/app.py CHANGED
@@ -30,6 +30,8 @@ def main():
30
  "The 'share' argument can only be set to 0 or 1, but was:",
31
  args.share,
32
  )
 
 
33
 
34
  # initialize and run app
35
  print("Launching demo...")
 
30
  "The 'share' argument can only be set to 0 or 1, but was:",
31
  args.share,
32
  )
33
+
34
+ print("Current cwd:", args.cwd)
35
 
36
  # initialize and run app
37
  print("Launching demo...")
demo/src/inference.py CHANGED
@@ -48,7 +48,7 @@ def run_model(
48
  # Setting up the configuration file
49
  rads_config = configparser.ConfigParser()
50
  rads_config.add_section("Default")
51
- rads_config.set("Default", "task", "neuro_diagnosis")
52
  rads_config.set("Default", "caller", "")
53
  rads_config.add_section("System")
54
  rads_config.set("System", "gpu_id", "-1")
 
48
  # Setting up the configuration file
49
  rads_config = configparser.ConfigParser()
50
  rads_config.add_section("Default")
51
+ rads_config.set("Default", "task", "mediastinum_diagnosis")
52
  rads_config.set("Default", "caller", "")
53
  rads_config.add_section("System")
54
  rads_config.set("System", "gpu_id", "-1")