Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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')
|