Update app.py
Browse files
app.py
CHANGED
|
@@ -77,6 +77,8 @@ def main():
|
|
| 77 |
|
| 78 |
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
|
| 79 |
|
|
|
|
|
|
|
| 80 |
embeddings = load_prep_data()
|
| 81 |
|
| 82 |
model = build_sent_trans_model()
|
|
|
|
| 77 |
|
| 78 |
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
|
| 79 |
|
| 80 |
+
st.info('Using device type: {}'.format(device))
|
| 81 |
+
|
| 82 |
embeddings = load_prep_data()
|
| 83 |
|
| 84 |
model = build_sent_trans_model()
|