Update Models/bert_strim.py

Models/bert_strim.py

@@ -31,10 +31,6 @@ def predict_sentiment(text):
         output = model(input_ids, attention_mask)
         prediction = torch.round(output).cpu().numpy()[0][0]
         return prediction
-    if prediction == 1:
-        return "Позитивный отзыв 😀"
-    else:
-        return "Негативный отзыв 😟"
 
 def bert_model_page():
     st.title("Классификация отзывов")