Update app.py

app.py

@@ -210,9 +210,9 @@ def selectedCorpusForNextQuarterModel(x,quarter,year):
     chunksAttention_mask=[tokens["attention_mask"][0][r*splitSize:(r+1)*splitSize] for r in range(math.ceil(len(tokens["attention_mask"][0])/splitSize))]
     l=[]
     for idx in range(len(chunksInput_ids)):
-        l.append({"input_ids":torch.tensor([list(chunksInput_ids[idx])]).to("cuda"),
-         "token_type_ids":torch.tensor([list(chunksToken_type_ids[idx])]).to("cuda"),
-          "attention_mask":torch.tensor([list(chunksAttention_mask[idx])]).to("cuda")
+        l.append({"input_ids":torch.tensor([list(chunksInput_ids[idx])]),
+         "token_type_ids":torch.tensor([list(chunksToken_type_ids[idx])]),
+          "attention_mask":torch.tensor([list(chunksAttention_mask[idx])])
         })
       
     selectedTopics = ["Stock Movement", "Earnings", "IPO", "Stock Commentary", "Currencies", "M&A | Investments", "Financials", "Macro", "Analyst Update", "Company | Product News"]