Update app.py

app.py

@@ -33,7 +33,7 @@ def predict_sentiment(text, model):
         pred = lr_model.predict_proba(x)[0]
         return {"neg": pred[0], "pos": pred[1]}
     elif model == "custom BERT":
-        pred = F.softmax(bert_model(**bert_tokenizer("I love you", return_tensors="pt")).logits[0], dim=0).tolist()
+        pred = F.softmax(bert_model(**bert_tokenizer(text, return_tensors="pt")).logits[0], dim=0).tolist()
         return {"neg": pred[0], "pos": pred[1]}