Update app.py

app.py

@@ -401,9 +401,9 @@ def ask(symbol, weeks_before, withbasic):
   print(model.device)
     
   inputs = tokenizer(pt, return_tensors='pt')
-  inputs = {key: value.to(model.device) for key, value in inputs.items()}
   #inputs = {key: value.to(model.device) for key, value in inputs.items()}
-  #inputs = {key: value.to('cuda:0') for key, value in inputs.items()}
+  #inputs = {key: value.to(model.device) for key, value in inputs.items()}
+  inputs = {key: value.to('cuda:0') for key, value in inputs.items()}
   print("Inputs loaded onto devices.")
 
   res = model.generate(