Moditha24 commited on
Commit
58e3800
·
verified ·
1 Parent(s): 3c5b534

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ import torch.nn as nn
3
  from config import MLP
4
  import gradio as gr
5
 
6
- # Load the model
7
  model = MLP()
8
  model.load_state_dict(torch.load("pytorch_model.pth", map_location=torch.device("cpu")))
9
  model.eval()
 
3
  from config import MLP
4
  import gradio as gr
5
 
6
+ # Load the mode
7
  model = MLP()
8
  model.load_state_dict(torch.load("pytorch_model.pth", map_location=torch.device("cpu")))
9
  model.eval()