File size: 468 Bytes
7bbfc34 66eabd8 3c8cf6c 7bbfc34 3c8cf6c 66eabd8 3c8cf6c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 |
import gradio as gr
#from transformers import pipeline
from transformers import AutoModel
#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(int(num_ecgs))
return {"ecgs": predictions}
gr.Interface(
predict,
inputs="text",
outputs="text",
title="Generating ECGs",
).launch()
|