Update app.py

app.py

@@ -52,9 +52,9 @@ def generate_predictions(text):
 
         # Make predictions
         sentence_prediction = predict(loaded_model, input_loader)[0]
-        predictions.append(sentence + ":" + str(sentence_prediction))
+        predictions.append(sentence_prediction)
 
-    return predictions
+    return predictions[0]  # Return the first prediction instead of the whole list
 
 # Specify the device as CPU
 device = torch.device('cpu')