updated app

app.py

@@ -1,3 +1,17 @@
 import gradio as gr
+from transformers import pipeline
+from transformers import AutoModel
 
-gr.Interface.load("models/deepsynthbody/deepfake_ecg", trust_remote_code=True).launch()
+#pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
+model = AutoModel.from_pretrained("deepsynthbody/deepfake_ecg", trust_remote_code=True)
+
+def predict(num_ecgs):
+    predictions = model(num_ecgs)
+    return {"ecgs": predictions}
+
+gr.Interface(
+    predict,
+    inputs="text",
+    outputs="text",
+    title="Generating ECGs",
+).launch()