zhengr commited on
Commit
9777dd4
·
verified ·
1 Parent(s): 1502095

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -394,8 +394,8 @@ def ask(symbol, weeks_before, withbasic):
394
  info, pt = get_all_prompts_online(symbol=symbol, weeks_before=weeks_before, with_basics=withbasic)
395
  # print(info)
396
 
397
- device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
398
- model=model.to(device)
399
 
400
  print(model.device)
401
  inputs = tokenizer(pt, return_tensors='pt')
 
394
  info, pt = get_all_prompts_online(symbol=symbol, weeks_before=weeks_before, with_basics=withbasic)
395
  # print(info)
396
 
397
+ #device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
398
+ #model=model.to(device)
399
 
400
  print(model.device)
401
  inputs = tokenizer(pt, return_tensors='pt')