eswardivi commited on
Commit
fccbbf3
1 Parent(s): 0a148ea

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -25,7 +25,8 @@ else:
25
  device = torch.device('cpu')
26
  print("Using CPU")
27
 
28
- model = model.to(device)
 
29
  model = model.to_bettertransformer()
30
 
31
  @spaces.GPU
 
25
  device = torch.device('cpu')
26
  print("Using CPU")
27
 
28
+ # model = model.to(device)
29
+ # Dispatch Errors
30
  model = model.to_bettertransformer()
31
 
32
  @spaces.GPU