simonduerr commited on
Commit
21cbe60
1 Parent(s): 66de88c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -328,7 +328,7 @@ def predict(input_mol, style, contour_level, view_str, chains):
328
  chain_str = ""
329
  chain_dict = json.loads(chains)
330
 
331
- outputfile = input_mol.replace(".pdb",".svg")
332
  # sort keys in dict and add colors to chain_str
333
  for chain in sorted(chain_dict.keys()):
334
  chain_str += f" '{chain_dict[chain]}'"
 
328
  chain_str = ""
329
  chain_dict = json.loads(chains)
330
 
331
+ outputfile = os.path.basename(input_mol).replace(".pdb",".svg")
332
  # sort keys in dict and add colors to chain_str
333
  for chain in sorted(chain_dict.keys()):
334
  chain_str += f" '{chain_dict[chain]}'"