Jfink09 commited on
Commit
701c8cb
·
verified ·
1 Parent(s): 61c910c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -24,9 +24,7 @@ model = LaserPredictions(6, 32, 3)
24
  model.load_state_dict(torch.load('laser_prescription_model.pt'))
25
  model.eval() # Set the model to evaluation mode
26
 
27
- You're absolutely right! In real nomograms, the target prescription is often set to 0, which means the post-op values would naturally be close to zero. In this case, using dummy values of zero for the post-op inputs when making predictions should not significantly interfere with the model's performance.
28
- Given this insight, let's update the code to reflect this approach:
29
- pythonCopy codeimport gradio as gr
30
  import pandas as pd
31
  from sklearn.model_selection import train_test_split
32
  import torch
 
24
  model.load_state_dict(torch.load('laser_prescription_model.pt'))
25
  model.eval() # Set the model to evaluation mode
26
 
27
+ import gradio as gr
 
 
28
  import pandas as pd
29
  from sklearn.model_selection import train_test_split
30
  import torch