Update app.py

app.py

@@ -72,7 +72,9 @@ def infer(seed):
 
     data = data.round(decimals = 2)
 
-    return f"{round(pred.flatten()[0]*100, 2)}%", data.values.tolist()
+    #return f"{round(pred.flatten()[0]*100, 5)}%", data.values.tolist()
+    result = 'This is fraudulent transaction!' if pred.flatten()[0] > 0.5 else "This is real transaction!"
+    return result, data.values.tolist()
 
 # get the inputs
 inputs = [gr.Slider(minimum=0, maximum=3000, step=1, label='Choose a random number', value=5)]
@@ -87,7 +89,7 @@ output = [
           ]
 
 title = 'Imbalanced Classification with Tensorflow'
-description = 'Imbalanced Classifiication in predicting Credit card Fraud.'
+description = 'Imbalanced Classification in predicting Credit card Fraud.'
 
 article = "Author: <a href=\"https://huggingface.co/geninhu\">Nhu Hoang</a>. Based on this <a href=\"https://keras.io/examples/structured_data/imbalanced_classification/\">keras example</a> by <a href=\"https://twitter.com/fchollet\">fchollet.</a> HuggingFace Model <a href=\"https://huggingface.co/keras-io/imbalanced_classification\">here</a> "