Update app.py

app.py

@@ -169,7 +169,7 @@ def trial(text):
             "token_type_ids": torch.tensor(tokenized['token_type_ids'],dtype=torch.long),
             "attention_mask": torch.tensor(tokenized['attention_mask'],dtype = torch.long)
         }
-    inputs = {k:inputs[k].unsqueeze(0).to(device=config['device']) for k in inputs.keys()}
+    inputs = {k:inputs[k].unsqueeze(0) for k in inputs.keys()}
 
     if model(inputs).item()>=0.5:
         return "AI"