Update app.py
Browse files
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 |
-
|
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
|