vlbthambawita commited on
Commit
66eabd8
1 Parent(s): b46bda1
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,12 +1,12 @@
1
  import gradio as gr
2
- from transformers import pipeline
3
  from transformers import AutoModel
4
 
5
  #pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
6
  model = AutoModel.from_pretrained("deepsynthbody/deepfake_ecg", trust_remote_code=True)
7
 
8
  def predict(num_ecgs):
9
- predictions = model(num_ecgs)
10
  return {"ecgs": predictions}
11
 
12
  gr.Interface(
 
1
  import gradio as gr
2
+ #from transformers import pipeline
3
  from transformers import AutoModel
4
 
5
  #pipeline = pipeline(task="image-classification", model="julien-c/hotdog-not-hotdog")
6
  model = AutoModel.from_pretrained("deepsynthbody/deepfake_ecg", trust_remote_code=True)
7
 
8
  def predict(num_ecgs):
9
+ predictions = model(int(num_ecgs))
10
  return {"ecgs": predictions}
11
 
12
  gr.Interface(