Spaces:
Running
Running
Update app.py
Browse files
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]}'"
|