Spaces:
Sleeping
Sleeping
Minor documentation updates.
Browse files
README.md
CHANGED
@@ -31,4 +31,4 @@ pip install -r requirements.txt
|
|
31 |
```bash
|
32 |
./app.py
|
33 |
```
|
34 |
-
Then, connect a web browser to http://127.0.0.1:7860/ to use the application
|
|
|
31 |
```bash
|
32 |
./app.py
|
33 |
```
|
34 |
+
Then, connect a web browser to [http://127.0.0.1:7860/](http://127.0.0.1:7860/) to use the application
|
app.py
CHANGED
@@ -12,7 +12,7 @@ import torch
|
|
12 |
model = AutoModel.from_pretrained("deepsynthbody/deepfake_ecg", trust_remote_code=True)
|
13 |
|
14 |
def predict():
|
15 |
-
prediction = (model(1)[0].t()/1000) # to micro
|
16 |
|
17 |
lead_III = (prediction[1] - prediction[0]).unsqueeze(dim=0)
|
18 |
lead_aVR = ((prediction[0] + prediction[1])*(-0.5)).unsqueeze(dim=0)
|
|
|
12 |
model = AutoModel.from_pretrained("deepsynthbody/deepfake_ecg", trust_remote_code=True)
|
13 |
|
14 |
def predict():
|
15 |
+
prediction = (model(1)[0].t()/1000) # to micro volts
|
16 |
|
17 |
lead_III = (prediction[1] - prediction[0]).unsqueeze(dim=0)
|
18 |
lead_aVR = ((prediction[0] + prediction[1])*(-0.5)).unsqueeze(dim=0)
|