import numpy as np
from make_predictions import make_prediction
import gradio as gr

def predict(file):
    print("=====================================")
    ans = make_prediction(file.name)
    print(ans)
    print("=====================================")
    return {ans[0] :ans[1]}

iface = gr.Interface(predict, "file", "label", title="ECG Classification", description="Upload an ECG file to classify it as Normal or Abnormal", examples=[["data/faizan_r8.txt"]], share=True)
iface.launch()