Haitam03 commited on
Commit
c6a824e
·
verified ·
1 Parent(s): 3519daf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -201,7 +201,7 @@ def load_model_and_vocabs():
201
  dropout=0.3,
202
  upsample_factor=2
203
  ).to(device)
204
-
205
  # Load trained weights
206
  model.load_state_dict(torch.load('best_model.pth', map_location=device))
207
  model.eval()
 
201
  dropout=0.3,
202
  upsample_factor=2
203
  ).to(device)
204
+
205
  # Load trained weights
206
  model.load_state_dict(torch.load('best_model.pth', map_location=device))
207
  model.eval()