Update app.py

app.py

@@ -40,7 +40,7 @@ class Bert_model(nn.Module):
 # Load the model
 model_path = "model.pt"  # Replace with your actual model path
 state_dict = torch.load(model_path)
-device = "cpu"  # or "cpu" if GPU is not available
+device = torch.device("cuda:0")  # or "cpu" if GPU is not available
 
 # Instantiate the model architecture
 model = Bert_model(hidden_size=768, dropout_rate=0.1)  # Adjust the hidden size to match the saved model