Update app.py

app.py

@@ -1,11 +1,25 @@
 import gradio as gr
-from transformers import AutoModel
+from huggingface_hub import from_pretrained_keras
+import pandas as pd
+import numpy as np
 
-model = AutoModel.from_pretrained("keras-io/timeseries_transformer_classification")
 
+model = from_pretrained_keras("keras-io/timeseries_transformer_classification")
 
+def detect_issue(df):
+  df = pd.read_csv('sample.csv',header=None)
+  pred = model.predict(df)[0]
+  problem = 'No problem'
+  if(np.argmax(pred)==1):
+    problem = 'Engine problem'
+  return problem, pred[1]
+  
 
-iface = gr.Interface(fraud_detector,"dataframe",gr.outputs.Label(label="Fraud Level"))
+iface = gr.Interface(detect_issue,"dataframe",outputs=[
+        gr.outputs.Textbox(label="Engine issue"),
+        gr.outputs.Textbox(label="Engine issue score"),
+    ])
 
+examples = [["examples/sample.csv"]]
 
 iface.launch()