matanru commited on
Commit
a51608b
1 Parent(s): 8ab33d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -192,6 +192,7 @@ model = None
192
  @spaces.GPU(duration=30)
193
  def estimate(data):
194
  global model
 
195
  with torch.no_grad():
196
  # model_device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
197
  # data["img_q"].to(device=model_device)
 
192
  @spaces.GPU(duration=30)
193
  def estimate(data):
194
  global model
195
+ model.cuda()
196
  with torch.no_grad():
197
  # model_device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
198
  # data["img_q"].to(device=model_device)