Marek Bukowicki commited on
Commit
7544717
·
1 Parent(s): d0daecc

fix GUI for default models, update links to doi

Browse files
Files changed (2) hide show
  1. Readme.md +1 -1
  2. predict-gui.py +2 -1
Readme.md CHANGED
@@ -3,7 +3,7 @@ ShimNet is a data-driven AI solution to improve high-resolution nuclear magnetic
3
  distorted by the inhomogeneous magnetic field (less than optimal shimming). To use it, the experimental training data has to be collected (see **Data collection** below).
4
  Example data can also be downloaded (see below).
5
 
6
- Paper: [ShimNet: A neural network for post-acquisition improvement of NMR spectra distorted by magnetic-field inhomogeneity](https://pubs.acs.org/doi/full/10.1021/acs.jpcb.5c02632)
7
 
8
  Web service: [![Open in Hugging Face Spaces](https://huggingface.co/datasets/huggingface/badges/resolve/main/open-in-hf-spaces-md.svg)](https://huggingface.co/spaces/NMR-CeNT-UW/ShimNet)
9
 
 
3
  distorted by the inhomogeneous magnetic field (less than optimal shimming). To use it, the experimental training data has to be collected (see **Data collection** below).
4
  Example data can also be downloaded (see below).
5
 
6
+ Paper: [ShimNet: A neural network for post-acquisition improvement of NMR spectra distorted by magnetic-field inhomogeneity](https://doi.org/10.1021/acs.jpcb.5c02632)
7
 
8
  Web service: [![Open in Hugging Face Spaces](https://huggingface.co/datasets/huggingface/badges/resolve/main/open-in-hf-spaces-md.svg)](https://huggingface.co/spaces/NMR-CeNT-UW/ShimNet)
9
 
predict-gui.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import torch
2
  torch.set_grad_enabled(False)
3
  import numpy as np
@@ -132,7 +133,7 @@ def process_file(input_file, config_file, weights_file, input_spectrometer_frequ
132
  # Gradio app
133
  with gr.Blocks() as app:
134
  gr.Markdown("# ShimNet Spectra Correction")
135
- gr.Markdown("[ShimNet: A neural network for post-acquisition improvement of NMR spectra distorted by magnetic-field inhomogeneity](https://chemrxiv.org/engage/chemrxiv/article-details/67ef86686dde43c90860d315)")
136
  gr.Markdown("Upload your input file, configuration, and weights to process the NMR spectrum.")
137
 
138
  with gr.Row():
 
1
+ import os
2
  import torch
3
  torch.set_grad_enabled(False)
4
  import numpy as np
 
133
  # Gradio app
134
  with gr.Blocks() as app:
135
  gr.Markdown("# ShimNet Spectra Correction")
136
+ gr.Markdown("[ShimNet: A neural network for post-acquisition improvement of NMR spectra distorted by magnetic-field inhomogeneity](https://doi.org/10.1021/acs.jpcb.5c02632)")
137
  gr.Markdown("Upload your input file, configuration, and weights to process the NMR spectrum.")
138
 
139
  with gr.Row():