Update pages/✨second.py

pages/✨second.py

@@ -50,7 +50,7 @@ prediction = model.predict(user_features)
 inputs = tokenizer(user_review_clean, max_length=512, padding=True, truncation=True, return_tensors='pt')
 outputs = model_bert(**inputs)
 prediction_bert = torch.nn.functional.softmax(outputs.logits, dim=1)
-prediction_bert = torch.argmax(predicted, dim=1).numpy(
+prediction_bert = torch.argmax(predicted, dim=1).numpy()
 st.write("Comment by ML model:", user_review)
 
 if prediction == 0: