zhengr commited on
Commit
7e10e22
·
verified ·
1 Parent(s): 24ff6b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -33,7 +33,7 @@ model = PeftModel.from_pretrained(model, peft_model, device_map="cuda:0", offlo
33
  #device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
34
  device = 'cuda:0'
35
  model = model.to(device)
36
- #model = model.eval()
37
 
38
  # Inference Data
39
  # get company news online
 
33
  #device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
34
  device = 'cuda:0'
35
  model = model.to(device)
36
+ model = model.eval()
37
 
38
  # Inference Data
39
  # get company news online